Cleanup the visual when menu is collapsed