diff --git a/nerfstudio/viewer_beta/control_panel.py b/nerfstudio/viewer_beta/control_panel.py
index d93d33546..07f556dd4 100644
--- a/nerfstudio/viewer_beta/control_panel.py
+++ b/nerfstudio/viewer_beta/control_panel.py
@@ -270,9 +270,10 @@ class ControlPanel:
             step: the train step to set the model to
         """
         with self.viser_server.atomic(), self.stat_folder:
+            # TODO change to a .value call instead of remove() and add, this makes it jittery
             with self.viser_server.atomic():
                 self.markdown.remove()
-                self.markdown.content = f"Step: {step}"
+                self.markdown = self.viser_server.add_gui_markdown(f"Step: {step}")
 
     def update_output_options(self, new_options: List[str]):
         """