Fix removeEntity