Fix missing menu collapse