fix the menu bug