diff --git a/packages/backend/src/graphql/mutations/update-step.ts b/packages/backend/src/graphql/mutations/update-step.ts
index 5d0b6cba..653568ba 100644
--- a/packages/backend/src/graphql/mutations/update-step.ts
+++ b/packages/backend/src/graphql/mutations/update-step.ts
@@ -31,12 +31,14 @@ const updateStepResolver = async (
     })
     .throwIfNotFound();
 
-  step = await Step.query().patchAndFetchById(input.id, {
-    key: input.key,
-    appKey: input.appKey,
-    connectionId: input.connection.id,
-    parameters: input.parameters,
-  });
+  step = await Step.query()
+    .patchAndFetchById(input.id, {
+      key: input.key,
+      appKey: input.appKey,
+      connectionId: input.connection.id,
+      parameters: input.parameters,
+    })
+    .withGraphFetched('connection');
 
   return step;
 };