Collapse menu fix