Fixed a bug regarding the menu