diff --git a/python/tools/benchmark.py b/python/tools/benchmark.py
index 21a32956..accf8311 100755
--- a/python/tools/benchmark.py
+++ b/python/tools/benchmark.py
@@ -23,9 +23,6 @@ import signal
 import argparse
 import time
 
-import numpy as np
-import matplotlib.pyplot as plt
-
 from dht.network import DhtNetwork
 from dht.network import DhtNetworkSubProcess
 from dht.tests import PerformanceTest, PersistenceTest
diff --git a/python/tools/dht/tests.py b/python/tools/dht/tests.py
index 41c4a981..46f984b7 100644
--- a/python/tools/dht/tests.py
+++ b/python/tools/dht/tests.py
@@ -7,6 +7,9 @@ import random
 import string
 import time
 
+import numpy as np
+import matplotlib.pyplot as plt
+
 from opendht import *
 from dht.network import DhtNetwork, DhtNetworkSubProcess
 
@@ -33,6 +36,7 @@ def reset_before_test(featureTestMethod):
         return featureTestMethod(*args, **kwargs)
     return call
 
+
 class FeatureTest(object):
     done = 0
     lock = None