Fixed removeEntity