fix bug return to menu