Fix bug in menu