diff --git a/src/blenderbim/test/core/test_drawing.py b/src/blenderbim/test/core/test_drawing.py
index 2b9060c516..2cadacdc26 100644
--- a/src/blenderbim/test/core/test_drawing.py
+++ b/src/blenderbim/test/core/test_drawing.py
@@ -442,7 +442,7 @@ class TestUpdateDrawingName:
         drawing.get_drawing_group("drawing").should_be_called().will_return("group")
         drawing.get_name("group").should_be_called().will_return("name")
         drawing.get_drawing_collection("drawing").should_be_called().will_return("collection")
-        drawing.set_drawing_collection_name("group", "collection").should_be_called()
+        drawing.set_drawing_collection_name("drawing", "collection").should_be_called()
 
         drawing.get_drawing_document("drawing").should_be_called().will_return("reference")
         drawing.get_reference_document("reference").should_be_called().will_return("information")
@@ -459,7 +459,7 @@ class TestUpdateDrawingName:
         drawing.get_name("group").should_be_called().will_return("oldname")
         ifc.run("attribute.edit_attributes", product="group", attributes={"Name": "name"}).should_be_called()
         drawing.get_drawing_collection("drawing").should_be_called().will_return("collection")
-        drawing.set_drawing_collection_name("group", "collection").should_be_called()
+        drawing.set_drawing_collection_name("drawing", "collection").should_be_called()
 
         drawing.get_drawing_document("drawing").should_be_called().will_return("reference")
         drawing.get_reference_document("reference").should_be_called().will_return("information")
diff --git a/src/blenderbim/test/core/test_material.py b/src/blenderbim/test/core/test_material.py
index c5856927ab..f931c5f192 100644
--- a/src/blenderbim/test/core/test_material.py
+++ b/src/blenderbim/test/core/test_material.py
@@ -161,4 +161,4 @@ class TestSelectByMaterial:
     def test_run(self, material, spatial):
         material.get_elements_by_material("material").should_be_called().will_return("elements")
         spatial.select_products("elements").should_be_called()
-        subject.select_by_material(material, material="material")
+        subject.select_by_material(material, spatial, material="material")
diff --git a/src/blenderbim/test/core/test_style.py b/src/blenderbim/test/core/test_style.py
index d9ccb1aed1..ac1d4dec04 100644
--- a/src/blenderbim/test/core/test_style.py
+++ b/src/blenderbim/test/core/test_style.py
@@ -218,4 +218,4 @@ class TestSelectByStyle:
     def test_run(self, style, spatial):
         style.get_elements_by_style("style").should_be_called().will_return("elements")
         spatial.select_products("elements").should_be_called()
-        subject.select_by_style(style, style="style")
+        subject.select_by_style(style, spatial, style="style")