Fix collapsed menu bug