diff --git a/src/bonsai/test/core/test_type.py b/src/bonsai/test/core/test_type.py
index 031e414f95..85e6aecd76 100644
--- a/src/bonsai/test/core/test_type.py
+++ b/src/bonsai/test/core/test_type.py
@@ -23,7 +23,12 @@ from test.core.bootstrap import geometry, ifc, model, type
 class TestAssignType:
     def test_assigning_and_switching_to_an_existing_type_data(self, ifc, model, type):
         type.record_material_usage_attributes("element").should_be_called().will_return(None)
-        ifc.run("type.assign_type", related_objects=["element"], relating_type="type").should_be_called()
+        ifc.run(
+            "type.assign_type",
+            related_objects=["element"],
+            relating_type="type",
+            should_map_representations=True,
+        ).should_be_called()
         model.get_usage_type("type").should_be_called(2).will_return(None)
         ifc.get_object("type").should_be_called().will_return("type_obj")
         type.get_object_data("type_obj").should_be_called().will_return("type_obj_data")
@@ -34,7 +39,12 @@ class TestAssignType:
 
     def test_assigning_and_not_changing_data_if_the_type_has_no_data(self, ifc, model, type):
         type.record_material_usage_attributes("element").should_be_called().will_return(None)
-        ifc.run("type.assign_type", related_objects=["element"], relating_type="type").should_be_called()
+        ifc.run(
+            "type.assign_type",
+            related_objects=["element"],
+            relating_type="type",
+            should_map_representations=True,
+        ).should_be_called()
         model.get_usage_type("type").should_be_called(2).will_return(None)
         ifc.get_object("type").should_be_called().will_return("type_obj")
         type.get_object_data("type_obj").should_be_called().will_return(None)