Fixed the falling bug