Fix menu collapse error