diff --git a/repositories/magic_plan/magic_plan_postgres_repository.py b/repositories/magic_plan/magic_plan_postgres_repository.py
index 1cd77a32..f8ff123c 100644
--- a/repositories/magic_plan/magic_plan_postgres_repository.py
+++ b/repositories/magic_plan/magic_plan_postgres_repository.py
@@ -6,7 +6,15 @@ from sqlalchemy import delete, select
 from sqlalchemy.dialects.postgresql import insert as pg_insert
 from sqlmodel import Session, col
 
-from domain.magicplan.models import Floor, Plan
+from domain.magicplan.models import (
+    Door,
+    DoorVentilation,
+    Floor,
+    Plan,
+    Room,
+    Window,
+    WindowVentilation,
+)
 from infrastructure.postgres.magic_plan_tables import (
     MagicPlanDoorModel,
     MagicPlanDoorVentilationModel,