dpkp
diff --git a/‎test/integration/__init__.py b/‎test/integration/__init__.py
diff --git a/‎test/test_sender.py
+4-1 b/‎test/test_sender.py
+4-1
@@ -6,7 +6,10 @@
 import time
 
 import pytest
-from unittest.mock import call
+try:
+    from unittest.mock import call
+except ImportError:
+    from mock import call
 
 from kafka.vendor import six