diff --git a/test/integration/test_workflow_refactoring.py b/test/integration/test_workflow_refactoring.py
index 17ba3819c60a..6bce5f080a6f 100644
--- a/test/integration/test_workflow_refactoring.py
+++ b/test/integration/test_workflow_refactoring.py
@@ -120,6 +120,7 @@ def test_basic_refactoring_types(self):
         assert self._latest_workflow.step_by_label("new_param2").position["left"] == 2
 
         assert len(self._latest_workflow.step_by_label("first_cat").inputs) == 1
+        assert len(self._latest_workflow.step_by_label("first_cat").inputs[0].connections) == 1
         actions = [
             {
                 "action_type": "disconnect",
@@ -128,7 +129,8 @@ def test_basic_refactoring_types(self):
             }
         ]
         self._refactor(actions)
-        assert len(self._latest_workflow.step_by_label("first_cat").inputs) == 0
+        assert len(self._latest_workflow.step_by_label("first_cat").inputs) == 1
+        assert len(self._latest_workflow.step_by_label("first_cat").inputs[0].connections) == 0
 
         actions = [
             {