Fixed a problems with menu