tweaks

1 files changed, 3 insertions, 1 deletions

diff --git a/src/sim/sim_manager/mod.rs b/src/sim/sim_manager/mod.rs
index 8cb5eb3..8849a1d 100644
--- a/src/sim/sim_manager/mod.rs
+++ b/src/sim/sim_manager/mod.rs
@@ -276,8 +276,10 @@ impl SimManager {
             self.cell_manager = CellManager::from_default_size();
         }
         if input_manager.pressed(Input::ClearEntities) {
-            let entities: Vec<EntityId> = self.entities.drain().map(|(id, _)| id).collect();
+            // TODO drain doesn't work here?
+            let entities: Vec<EntityId> = self.entities.keys().cloned().collect();
             entities.iter().for_each(|&id| self.destroy_entity(id));
+            self.entities.clear();
         }
         if input_manager.pressed(Input::ClearParticles) {
             self.particle_manager.particles = Vec::new();