Fixed a small bug in menu