Trying to fix bug when collapsing menu