diff --git a/aiidalab_ispg/app/steps.py b/aiidalab_ispg/app/steps.py
index f373d1c..cbc7803 100644
--- a/aiidalab_ispg/app/steps.py
+++ b/aiidalab_ispg/app/steps.py
@@ -164,7 +164,7 @@ def __init__(self, progress_bar, children=None, **kwargs):
                 self._update_step_state,
                 self._update_workflow_state,
             ],
-            on_sealed=[self._display_results],
+            on_sealed=[self._display_results, self._update_kill_button],
         )
         ipw.dlink((self, "process_uuid"), (self.process_monitor, "value"))