Fixed small bug in menu