Fixed Breaking Legacy change

widgets/agentmoveoperations.py

@@ -506,6 +506,7 @@ class AgentMoveOperations(Widget):
         unsuccessful = []
         status_label = self.query_one("#status_label", Static)
         status_label.update("Exporting CSV...")
+        self.app.refresh_data()
         agents = self.agents
         policies = self.app.policies
         path = self.app.working_dir
@@ -604,6 +605,8 @@ class AgentMoveOperations(Widget):
 
                     successful.append((agent, f"Moved to {mode}: {result}"))
                     logger.info(f"Successfully toggled {agent.hostname} to {mode}")
+                    self.app.refresh_data()
+
                 except Exception as e:
                     unsuccessful.append((agent, str(e)))
                     logger.error(f"Failed to toggle {agent.hostname}: {e}")
@@ -728,7 +731,7 @@ class AgentMoveOperations(Widget):
             status_label.update(f"Error: {str(e)}")
             self.operation_in_progress = False
             return
-
+        self.app.refresh_data()
         self.operation_in_progress = False
         status_label.update("Operation complete!")