1 files changed, 3 insertions, 2 deletions
diff --git a/test/testutil.py b/test/testutil.py
index dc8eea0..e7dcda6 100644
--- a/test/testutil.py
+++ b/test/testutil.py
@@ -5,9 +5,10 @@ import random
 import socket
 import string
 import time
-import unittest2
 import uuid
 
+from . import unittest
+
 from kafka.common import OffsetRequest
 from kafka import KafkaClient
 
@@ -45,7 +46,7 @@ def get_open_port():
     sock.close()
     return port
 
-class KafkaIntegrationTestCase(unittest2.TestCase):
+class KafkaIntegrationTestCase(unittest.TestCase):
     create_client = True
     topic = None
     server = None