diff --git a/src/bonsai/test/bim/feature/attribute.feature b/src/bonsai/test/bim/feature/attribute.feature
index 04ef05a201..a1af4d722f 100644
--- a/src/bonsai/test/bim/feature/attribute.feature
+++ b/src/bonsai/test/bim/feature/attribute.feature
@@ -19,7 +19,7 @@ Scenario: Edit attributes
     And the object "IfcSite/My Site" is selected
     And I press "bim.enable_editing_attributes(obj='IfcSite/My Site')"
     And I set "active_object.BIMAttributeProperties.attributes[1].string_value" to "Name"
-    When I press "bim.edit_attributes(obj='IfcSite/My Site')"
+    When I press "bim.edit_attributes"
     Then the object "IfcSite/Name" is an "IfcSite"
     And the object "IfcSite/My Site" does not exist
 
@@ -29,9 +29,9 @@ Scenario: Edit attributes - longitude / latitude
     And I press "bim.enable_editing_attributes(obj='IfcSite/My Site')"
     Then "active_object.BIMAttributeProperties.attributes[6].name" is "RefLatitude"
     When I set "active_object.BIMAttributeProperties.attributes[6].string_value" to "[1,2]"
-    And I press "bim.edit_attributes(obj='IfcSite/My Site')"
+    And I press "bim.edit_attributes"
     And I press "bim.enable_editing_attributes(obj='IfcSite/My Site')"
-    And I press "bim.edit_attributes(obj='IfcSite/My Site')"
+    And I press "bim.edit_attributes"
     Then nothing happens
 
 Scenario: Copy attribute to selected
diff --git a/src/bonsai/test/bim/feature/demo.feature b/src/bonsai/test/bim/feature/demo.feature
index 096eae8f99..628982c16c 100644
--- a/src/bonsai/test/bim/feature/demo.feature
+++ b/src/bonsai/test/bim/feature/demo.feature
@@ -1,3 +1,6 @@
+@demo
+Feature: Demo
+
 # ############################################################################ #
 
 # Hey there! Welcome to the Bonsai code. Please feel free to reach
@@ -17,9 +20,6 @@
 #|# in test_feature.py. Most of the time, there is already a sentence defined for
 #|# what you want to test.
 #|
-#|@demo
-#|Feature: Demo
-#|
 #|# Every operator has at least one scenario associated with it to test it.
 #|Scenario: Demonstrate hello world
 #|    Given an empty IFC project