diff --git a/backend/documents_parser/tests/test_end_to_end.py b/backend/documents_parser/tests/test_end_to_end.py
index 359a4ea8..f35dfd94 100644
--- a/backend/documents_parser/tests/test_end_to_end.py
+++ b/backend/documents_parser/tests/test_end_to_end.py
@@ -183,12 +183,14 @@ class TestPdfToEpcPropertyData:
                             total_floor_area_m2=35.68,
                             party_wall_length_m=10.62,
                             heat_loss_perimeter_m=13.44,
+                            floor=1,
                         ),
                         SapFloorDimension(
                             room_height_m=2.17,
                             total_floor_area_m2=35.68,
                             party_wall_length_m=10.62,
                             heat_loss_perimeter_m=11.0,
+                            floor=0,
                         ),
                     ],
                     wall_thickness_mm=310,
@@ -206,6 +208,7 @@ class TestPdfToEpcPropertyData:
                             total_floor_area_m2=3.8,
                             party_wall_length_m=0.0,
                             heat_loss_perimeter_m=5.7,
+                            floor=0,
                         ),
                     ],
                     wall_thickness_mm=310,
diff --git a/datatypes/epc/domain/tests/test_from_site_notes.py b/datatypes/epc/domain/tests/test_from_site_notes.py
index 81efe32d..6ae4c851 100644
--- a/datatypes/epc/domain/tests/test_from_site_notes.py
+++ b/datatypes/epc/domain/tests/test_from_site_notes.py
@@ -464,12 +464,14 @@ class TestFromSiteNotesExample1:
                             total_floor_area_m2=24.78,
                             party_wall_length_m=6.15,
                             heat_loss_perimeter_m=14.21,
+                            floor=1,
                         ),
                         SapFloorDimension(
                             room_height_m=2.35,
                             total_floor_area_m2=24.78,
                             party_wall_length_m=6.15,
                             heat_loss_perimeter_m=14.21,
+                            floor=0,
                         ),
                     ],
                     wall_thickness_mm=280,