diff --git a/erynrl/map/generator/room.py b/erynrl/map/generator/room.py
index 9a3423d..d3a90c5 100644
--- a/erynrl/map/generator/room.py
+++ b/erynrl/map/generator/room.py
@@ -9,6 +9,8 @@ import tcod
 
 from ... import log
 from ...geometry import Point, Rect, Size
+from ..room import RectangularRoom
+from ..tile import Empty, Floor, StairsDown, StairsUp, Wall
 
 if TYPE_CHECKING:
     from .. import Map