Fix menu bugs