diff --git a/game/components.py b/game/components.py
index 5eae5c0..05554e5 100644
--- a/game/components.py
+++ b/game/components.py
@@ -65,7 +65,7 @@ class Graphic:
     ch: int = ord("!")
     fg: tuple[int, int, int] = (255, 255, 255)
 
-Action: Final = ("Action", Callable[[tcod.ecs.Entity], None])
+Action: Final = ("Action", Callable[[Entity], None])
 """Possible action."""
 
 Gold: Final = ("Gold", int)