diff --git a/src/components/dialog/footer/ManagerProgressFooter.vue b/src/components/dialog/footer/ManagerProgressFooter.vue
index 420645b62..654c0c03a 100644
--- a/src/components/dialog/footer/ManagerProgressFooter.vue
+++ b/src/components/dialog/footer/ManagerProgressFooter.vue
@@ -70,6 +70,7 @@ import { useI18n } from 'vue-i18n'
 
 import { api } from '@/scripts/api'
 import { useComfyManagerService } from '@/services/comfyManagerService'
+import { useWorkflowService } from '@/services/workflowService'
 import {
   useComfyManagerStore,
   useManagerProgressDialogStore
@@ -96,14 +97,21 @@ const currentTaskName = computed(() => {
 })
 
 const handleRestart = async () => {
-  await useComfyManagerService().rebootComfyUI()
-  closeDialog()
-
   const onReconnect = async () => {
-    await useCommandStore().execute('Comfy.RefreshNodeDefinitions')
+    // Refresh manager state
+
     comfyManagerStore.clearLogs()
     comfyManagerStore.setStale()
+
+    // Refresh node definitions
+    await useCommandStore().execute('Comfy.RefreshNodeDefinitions')
+
+    // Reload workflow
+    await useWorkflowService().reloadCurrentWorkflow()
   }
   useEventListener(api, 'reconnected', onReconnect, { once: true })
+
+  await useComfyManagerService().rebootComfyUI()
+  closeDialog()
 }
 </script>