Test fixes after flake8 run

2 files changed, 4 insertions, 3 deletions

diff --git a/kafka/client.py b/kafka/client.py
index a31ea66..965cbc5 100644
--- a/kafka/client.py
+++ b/kafka/client.py
@@ -1,9 +1,10 @@
 from collections import defaultdict
 from functools import partial
+from itertools import count
 import logging
 import time
 
-from kafka.common import count, ErrorMapping, TopicAndPartition
+from kafka.common import ErrorMapping, TopicAndPartition
 from kafka.conn import KafkaConnection
 from kafka.protocol import KafkaProtocol
 
diff --git a/test/test_unit.py b/test/test_unit.py
index c796c94..3f3af66 100644
--- a/test/test_unit.py
+++ b/test/test_unit.py
@@ -3,7 +3,8 @@ import random
 import struct
 import unittest
 
-from kafka.client import KafkaClient, ProduceRequest, FetchRequest
+from kafka.client import KafkaClient
+from kafka.common import ProduceRequest, FetchRequest
 from kafka.codec import (
     has_gzip, has_snappy,
     gzip_encode, gzip_decode,
@@ -59,7 +60,6 @@ class TestMisc(unittest.TestCase):
     def test_length_prefix(self):
         for i in xrange(ITERATIONS):
             s1 = random_string()
-            s2 = length_prefix_message(s1)
             self.assertEquals(struct.unpack('>i', s2[0:4])[0], len(s1))