1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_conn.py b/test/test_conn.py
index 5451398..bc644d7 100644
--- a/test/test_conn.py
+++ b/test/test_conn.py
@@ -2,12 +2,12 @@ import socket
 import struct
 
 import mock
-import unittest2
+from . import unittest
 
 from kafka.common import *
 from kafka.conn import *
 
-class ConnTest(unittest2.TestCase):
+class ConnTest(unittest.TestCase):
     def setUp(self):
         self.config = {
             'host': 'localhost',