Fixig bug in menu