diff --git a/lib/galaxy_test/api/test_workflows.py b/lib/galaxy_test/api/test_workflows.py
index 3317ae550a38..d29ba34de013 100644
--- a/lib/galaxy_test/api/test_workflows.py
+++ b/lib/galaxy_test/api/test_workflows.py
@@ -5012,7 +5012,11 @@ def test_run_with_int_parameter(self):
             content = self.dataset_populator.get_history_dataset_content(history_id)
             assert len(content.splitlines()) == 1, content
             invocation = self.workflow_populator.get_invocation(run_response.invocation_id)
-            assert invocation["input_step_parameters"]["int_input"]["parameter_value"] == 1
+            parameter_step = next(
+                (step for step in invocation["steps"] if step.get("workflow_step_label") == "int_input"), None
+            )
+            parameter_step_id = parameter_step["workflow_step_id"]
+            assert invocation["input_step_parameters"][parameter_step_id]["parameter_value"] == 1
 
             run_response = self._run_workflow(
                 WORKFLOW_PARAMETER_INPUT_INTEGER_OPTIONAL,
@@ -5027,7 +5031,7 @@ def test_run_with_int_parameter(self):
             )
             invocation = self.workflow_populator.get_invocation(run_response.invocation_id)
             # Optional step parameter without default value will not be recorded.
-            assert "int_input" not in invocation["input_step_parameters"]
+            assert not invocation["input_step_parameters"]
 
     def test_run_with_int_parameter_nested(self):
         with self.dataset_populator.test_history() as history_id: