Fixed a bug with menu