diff --git a/app/src/Dashboard.svelte b/app/src/Dashboard.svelte
index c865c91d..e74e4481 100644
--- a/app/src/Dashboard.svelte
+++ b/app/src/Dashboard.svelte
@@ -162,7 +162,7 @@
             on:start_message={removeStopClass}
           />
 
-          <NodeUpdate />
+          <NodeUpdate {updateBody} />
 
           <RestartNode />
         {/if}
diff --git a/app/src/nodes/NodeUpdate.svelte b/app/src/nodes/NodeUpdate.svelte
index fa49c137..70f4b8d7 100644
--- a/app/src/nodes/NodeUpdate.svelte
+++ b/app/src/nodes/NodeUpdate.svelte
@@ -6,6 +6,7 @@
   import { getImageVersion } from "../helpers/swarm";
 
   let updating = false;
+  export let updateBody = () => {};
 
   async function upgradeContainer() {
     let name = $selectedNode.name;
@@ -13,6 +14,7 @@
     updating = true;
     await api.swarm.update_node(name);
     await getImageVersion(stack, selectedNode);
+    updateBody();
     updating = false;
   }
 </script>