diff --git a/maths/segmented_sieve.py b/maths/segmented_sieve.py
index 0054b0595be5..7fddacd44c7e 100644
--- a/maths/segmented_sieve.py
+++ b/maths/segmented_sieve.py
@@ -3,7 +3,7 @@
 import math
 
 
-def sieve(n):
+def sieve(n: int) -> list:
     """Segmented Sieve."""
     in_prime = []
     start = 2