KAFKA-992 follow up. Fix broken unit test