diff --git a/src/blenderbim/blenderbim/bim/module/resource/operator.py b/src/blenderbim/blenderbim/bim/module/resource/operator.py
index 436b4e2906..1566bfd4fd 100644
--- a/src/blenderbim/blenderbim/bim/module/resource/operator.py
+++ b/src/blenderbim/blenderbim/bim/module/resource/operator.py
@@ -145,7 +145,7 @@ class AssignResource(bpy.types.Operator, tool.Ifc.Operator):
     related_object: bpy.props.StringProperty()
 
     def _execute(self, context):
-        core.assign_resource(tool.Ifc, tool.Resource, resource=tool.Ifc.get().by_id(self.resource))
+        core.assign_resource(tool.Ifc, tool.Spatial, resource=tool.Ifc.get().by_id(self.resource))
 
 
 class UnassignResource(bpy.types.Operator, tool.Ifc.Operator):
@@ -156,7 +156,7 @@ class UnassignResource(bpy.types.Operator, tool.Ifc.Operator):
     related_object: bpy.props.StringProperty()
 
     def _execute(self, context):
-        core.unassign_resource(tool.Ifc, tool.Resource, resource=tool.Ifc.get().by_id(self.resource))
+        core.unassign_resource(tool.Ifc, tool.Spatial, resource=tool.Ifc.get().by_id(self.resource))
 
 
 class EnableEditingResourceTime(bpy.types.Operator, tool.Ifc.Operator):
diff --git a/src/blenderbim/blenderbim/core/resource.py b/src/blenderbim/blenderbim/core/resource.py
index 596501afb4..840a6c24d6 100644
--- a/src/blenderbim/blenderbim/core/resource.py
+++ b/src/blenderbim/blenderbim/core/resource.py
@@ -158,15 +158,15 @@ def contract_resource(resource_tool, resource):
     resource_tool.load_resource_properties()
 
 
-def assign_resource(ifc, resource_tool, resource=None, products=None):
+def assign_resource(ifc, spatial, resource=None, products=None):
     if not products:
-        products = resource_tool.get_selected_products()
+        products = spatial.get_selected_products()
     for product in products:
         rel = ifc.run("resource.assign_resource", relating_resource=resource, related_object=product)
 
 
-def unassign_resource(ifc, resource_tool, resource=None, products=None):
+def unassign_resource(ifc, spatial, resource=None, products=None):
     if not products:
-        products = resource_tool.get_selected_products()
+        products = spatial.get_selected_products()
     for product in products:
         ifc.run("resource.unassign_resource", relating_resource=resource, related_object=product)