diff --git a/kafka_consumer/tests/test_kafka_consumer.py b/kafka_consumer/tests/test_kafka_consumer.py index a62410a48d1393..e4e2690e2d0ff6 100644 --- a/kafka_consumer/tests/test_kafka_consumer.py +++ b/kafka_consumer/tests/test_kafka_consumer.py @@ -223,7 +223,7 @@ def test_version_metadata(datadog_agent, kafka_instance, dd_run_check): kafka_consumer_check = KafkaCheck('kafka_consumer', {}, [kafka_instance]) kafka_consumer_check.check_id = 'test:123' - kafka_client = kafka_consumer_check.create_kafka_client() + kafka_client = kafka_consumer_check.kafka_client version_data = [str(part) for part in kafka_client.check_version()] kafka_client.close() version_parts = {'version.{}'.format(name): part for name, part in zip(('major', 'minor', 'patch'), version_data)}