Fixed a little bug in the collision