Fixed the movement bug