Test clicking on the screen

squirrelbattle/entities/items.py

@@ -28,7 +28,7 @@ class Item(Entity):
         """
         if self.held:
             self.held_by.inventory.remove(self)
-            self.map.add_entity(self)
+            self.held_by.map.add_entity(self)
             self.move(self.held_by.y, self.held_by.x)
             self.held = False
             self.held_by = None
@@ -49,7 +49,7 @@ class Item(Entity):
         """
         self.held = True
         self.held_by = player
-        self.map.remove_entity(self)
+        self.held_by.map.remove_entity(self)
         player.inventory.append(self)
 
     def save_state(self) -> dict: