Corrected bug with menu