Fix menu collapse