Fixed the bug in the menu button