diff --git a/sw/decoders/maximum_likelihood.py b/sw/decoders/maximum_likelihood.py
index f78e7d2..1871069 100644
--- a/sw/decoders/maximum_likelihood.py
+++ b/sw/decoders/maximum_likelihood.py
@@ -1,4 +1,3 @@
-import numpy
 import numpy as np
 import itertools
 
@@ -48,4 +47,4 @@ class MLDecoder:
         """
         correlations = np.dot(self._codewords_bpsk, y)
 
-        return self._codewords[numpy.argmax(correlations)]
+        return self._codewords[np.argmax(correlations)]