diff --git a/puzzles/day02.py b/puzzles/day02.py
index 4f68c94..245b734 100644
--- a/puzzles/day02.py
+++ b/puzzles/day02.py
@@ -28,7 +28,7 @@ class Location:
 
         return self.calculate()
 
-    def calculate(self):
+    def calculate(self) -> int:
         return self.horizontal * self.depth
 
 
@@ -65,7 +65,7 @@ def part2(inputs: List[int]) -> int:
     return puzzle.run(inputs)
 
 
-def parse(inputs: str) -> List[List[str]]:
+def parse(inputs: str) -> List[str]:
     """Parse the input string"""
     return inputs.split("\n")