diff --git a/src/bonsai/test/bim/feature/search.feature b/src/bonsai/test/bim/feature/search.feature
index 7bbc05f983..807da1722a 100644
--- a/src/bonsai/test/bim/feature/search.feature
+++ b/src/bonsai/test/bim/feature/search.feature
@@ -24,9 +24,12 @@ Scenario: Edit filter query
 
 Scenario: Colour by property - default class query
     Given an empty IFC project
+    And I load the demo construction library
+    And I set "scene.BIMModelProperties.ifc_class" to "IfcColumnType"
+    And I add the construction type
     When I look at the "Colour By Property" panel
     And I click "Colour by Property"
-    Then the "BIM_UL_colourscheme" list has 4 items
+    Then the "BIM_UL_colourscheme" list has 1 items
 
 Scenario: Colour by property - no query
     Given an empty IFC project
diff --git a/src/bonsai/test/bim/test_feature.py b/src/bonsai/test/bim/test_feature.py
index fc551cd07d..cca817418e 100644
--- a/src/bonsai/test/bim/test_feature.py
+++ b/src/bonsai/test/bim/test_feature.py
@@ -472,7 +472,7 @@ def i_dont_see_text(text):
 def i_dont_see_the_name_list(name):
     assert panel_spy
     panel_spy.refresh_spy()
-    assert name not in [l["listtype_name"] for l in panel_spy.spied_lists]
+    assert name not in [l.spied_data["listtype_name"] for l in panel_spy.spied_lists]
 
 
 @given(parsers.parse('I see the "{prop}" property'))