diff --git a/python/tools/benchmark.py b/python/tools/benchmark.py
index 21a32956ea36e85bac45487dae9bd6f407fff1b8..accf83111c80ff19e4cf8cdad8b2d07028451306 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 41c4a9817bea3b7c47cab62b65a6527c9838e303..46f984b717de7810f237a26fd8e270252428563e 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