Fixed bug for the square collision