Fix the menu display on small screens