diff --git a/shared/workspaces/test/test_capability_steps.py b/shared/workspaces/test/test_capability_steps.py
index 1ab24abe38bdd3dc147f38ec8d1bdde43a001515..f758654c71fbbd0df8bdd8b762445a317eda8eba 100644
--- a/shared/workspaces/test/test_capability_steps.py
+++ b/shared/workspaces/test/test_capability_steps.py
@@ -13,15 +13,15 @@ def test_parse_workflow():
     """
     Tests that CapabilityStep parsing works as intended
     """
-    step = CapabilityStep.from_str("prepare-and-run-workflow foo bar")
+    step = CapabilityStep.from_str("prepare-and-run-workflow foo [bar]")
     assert isinstance(step, PrepareAndRunWorkflow)
-    step = CapabilityStep.from_str("await-qa foo bar")
+    step = CapabilityStep.from_str("await-qa foo [bar]")
     assert isinstance(step, AwaitQa)
-    step = CapabilityStep.from_str("await-workflow foo bar")
+    step = CapabilityStep.from_str("await-workflow foo [bar]")
     assert isinstance(step, AwaitWorkflow)
-    step = CapabilityStep.from_str("await-product foo bar")
+    step = CapabilityStep.from_str("await-product foo [bar]")
     assert isinstance(step, AwaitProduct)
-    step = CapabilityStep.from_str("await-parameter foo bar")
+    step = CapabilityStep.from_str("await-parameter foo [bar]")
     assert isinstance(step, AwaitParameter)
-    step = CapabilityStep.from_str("await-large-alloc-approval foo bar")
+    step = CapabilityStep.from_str("await-large-alloc-approval foo [bar]")
     assert isinstance(step, AwaitLargeAllocationApproval)