fixed tests to work with latest sarama