fix bug on menu collapse