diff --git a/server/src/workflow-management/classes/Generator.ts b/server/src/workflow-management/classes/Generator.ts
index 2eb5f130..107dc500 100644
--- a/server/src/workflow-management/classes/Generator.ts
+++ b/server/src/workflow-management/classes/Generator.ts
@@ -708,6 +708,7 @@ export class WorkflowGenerator {
   public updateSocket = (socket: Socket): void => {
     this.socket = socket;
     this.registerEventHandlers(socket);
+    this.initializeSocketListeners();
   };
 
   /**