trivial follow-up patch for kafka-1073 to fix unit tests; patched by Jun Rao
KAFKA-1075; Consumer will not rebalance upon topic partition change; reviewed by Neha Narkhede and Jun Rao