diff --git a/src/blenderbim/test/tool/test_drawing.py b/src/blenderbim/test/tool/test_drawing.py
index 6e790e6632..86eecc143e 100644
--- a/src/blenderbim/test/tool/test_drawing.py
+++ b/src/blenderbim/test/tool/test_drawing.py
@@ -305,6 +305,9 @@ class TestGetDrawingCollection(NewFile):
         collection = bpy.data.collections.new("Collection")
         bpy.context.scene.collection.children.link(collection)
         collection.objects.link(obj)
+        obj.BIMObjectProperties.collection = collection
+        collection.BIMCollectionProperties.obj = obj
+
         element = ifc.createIfcAnnotation()
         tool.Ifc.link(element, obj)
         assert subject.get_drawing_collection(element) == collection
diff --git a/src/blenderbim/test/tool/test_project.py b/src/blenderbim/test/tool/test_project.py
index 29e1891858..1cbd4a4cfb 100644
--- a/src/blenderbim/test/tool/test_project.py
+++ b/src/blenderbim/test/tool/test_project.py
@@ -44,7 +44,7 @@ class TestCreateEmpty(NewFile):
 
 class TestLoadDefaultThumbnails(NewFile):
     def test_nothing(self):
-        pass # Not possible to test this headlessly
+        pass  # Not possible to test this headlessly
 
 
 class TestRunAggregateAssignObject(NewFile):
@@ -93,6 +93,9 @@ class TestSetActiveSpatialElement(NewFile):
         collection = bpy.data.collections.new("Foo")
         bpy.context.scene.collection.children.link(collection)
         collection.objects.link(obj)
+        obj.BIMObjectProperties.collection = collection
+        collection.BIMCollectionProperties.obj = obj
+
         layer = bpy.context.view_layer.layer_collection.children["Foo"]
         assert bpy.context.view_layer.active_layer_collection != layer
         subject.set_active_spatial_element(obj)