diff --git a/test/integration/test_workflow_tasks.py b/test/integration/test_workflow_tasks.py
index c250b624439f..939e83b5b637 100644
--- a/test/integration/test_workflow_tasks.py
+++ b/test/integration/test_workflow_tasks.py
@@ -112,7 +112,7 @@ def _test_export_import_invocation_with_input_as_output(self, use_uris):
             out = output_values["wf_output_param"]
             assert out == "A text variable"
             inputs = invocation_details["input_step_parameters"]
-            assert "text_input" in inputs
+            assert list(inputs.values())[0]["label"] == "text_input"
 
             self._rerun_imported_workflow(summary, invocation_details)