diff --git a/backend/magic_plan/handler.py b/backend/magic_plan/handler.py
index 2629c810..a592cc6a 100644
--- a/backend/magic_plan/handler.py
+++ b/backend/magic_plan/handler.py
@@ -4,6 +4,7 @@ from backend.app.config import get_settings
 from backend.magic_plan.magic_plan_client import MagicPlanClient
 from backend.magic_plan.magic_plan_service import MagicPlanService
 from backend.magic_plan.magic_plan_trigger_request import MagicPlanTriggerRequest
+from datatypes.magicplan.domain.models import Plan
 from backend.utils.subtasks import task_handler
 from utils.logger import setup_logger
 
@@ -18,11 +19,18 @@ def handler(body: dict[str, Any], context: Any) -> str:
         customer_id=settings.MAGICPLAN_CUSTOMER_ID,
         api_key=settings.MAGICPLAN_API_KEY,
     )
-    plan = MagicPlanService(client).run(payload.address, payload.uprn)
+    plan: Plan = MagicPlanService(client).run(payload.address, payload.uprn)
     logger.info("Saved MagicPlan plan uid=%s", plan.uid)
     return plan.uid
 
 
 if __name__ == "__main__":
-    event = {"Records": [{"body": '{"address": "2 Laburnum Way Bromley BR2 8BZ"}', "messageId": "local-test"}]}
+    event = {
+        "Records": [
+            {
+                "body": '{"address": "2 Laburnum Way Bromley BR2 8BZ"}',
+                "messageId": "local-test",
+            }
+        ]
+    }
     handler(event, None)