diff --git a/src/logic/pieces/piece.py b/src/logic/pieces/piece.py
index cd44a3b..3e0d4cc 100644
--- a/src/logic/pieces/piece.py
+++ b/src/logic/pieces/piece.py
@@ -1,3 +1,4 @@
+from logic.board import Board
 from logic.position import Position
 from logic.move import Move
 from enum import Enum
@@ -17,5 +18,5 @@ class Piece:
     def position(self) -> Position:
         return self.pos
 
-    def legal_moves(self, board) -> list[Move]:
+    def legal_moves(self, board: Board) -> list[Move]:
         raise NotImplementedError(f"Can't say what the legal moves are for {type(self).__name__}, the method hasn't been implemented yet")