fixed a bug with the menu