diff --git a/datatypes/magicplan/domain/mapper.py b/datatypes/magicplan/domain/mapper.py
index c841a963..296acc9d 100644
--- a/datatypes/magicplan/domain/mapper.py
+++ b/datatypes/magicplan/domain/mapper.py
@@ -1,3 +1,4 @@
+import datatypes.magicplan.api.response as api
 from datatypes.magicplan.api.response import MagicPlan
 from datatypes.magicplan.domain.models import Plan, Floor, Room, Window, Door
 
@@ -10,7 +11,7 @@ def map_plan(mp: MagicPlan) -> Plan:
     )
 
 
-def _map_floor(f) -> Floor:
+def _map_floor(f: api.Floor) -> Floor:
     return Floor(
         level=f.level,
         name=f.name,
@@ -18,7 +19,7 @@ def _map_floor(f) -> Floor:
     )
 
 
-def _map_room(r) -> Room:
+def _map_room(r: api.Room) -> Room:
     width, length = _parse_dimensions(r.dimensions)
     return Room(
         name=r.name,
@@ -39,7 +40,7 @@ def _parse_dimensions(dimensions: str | None) -> tuple[float, float]:
     return width, length
 
 
-def _map_window(wi) -> Window:
+def _map_window(wi: api.WallItem) -> Window:
     return Window(
         width=round(wi.size.x, 2),
         height=round(wi.size.z, 2),
@@ -48,5 +49,5 @@ def _map_window(wi) -> Window:
     )
 
 
-def _map_door(wi) -> Door:
+def _map_door(wi: api.WallItem) -> Door:
     return Door(width=round(wi.size.x, 2))