Update examples/energy_expenditure/model.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: nyksans

examples/energy_expenditure/model.py

@@ -61,8 +61,10 @@ def rest(self):
             self.energy += 1
 
     def die(self):
-        """Safely remove agent from grid and model."""
-        self.model.grid.remove_agent(self)
+        """Safely remove agent from grid, model, and custom tracking."""
+        # First deregister from Mesa's model and grid tracking
+        self.remove()
+        # Then keep the model's custom agents_list in sync
         self.model.remove_agent(self)