Fix broker partition info logic introduced by hack 8c2369c45e85d1f8ba95e89f23fb7aaf4bc15731