logic fix for menu