fix in menu function