Fix broken onGround status when levitating