Fixing bug with menu