Fixed issue with the menu on small screens