Menu bug fix