Fix bug with jumping off screen