diff --git a/daras_ai_v2/base.py b/daras_ai_v2/base.py
index 0f1871cca..a06dbeddb 100644
--- a/daras_ai_v2/base.py
+++ b/daras_ai_v2/base.py
@@ -633,6 +633,11 @@ def _render_publish_modal(
                 gui.error(str(e))
                 return
 
+        if self._has_current_run_changed(current_run):
+            sr = self._on_submit()
+            if sr:
+                current_run = sr
+
         if is_update_mode:
             updates = dict(
                 saved_run=current_run,