diff --git a/lazuardy_anfis/tests.py b/lazuardy_anfis/tests.py
index 96cf8c1..329da49 100644
--- a/lazuardy_anfis/tests.py
+++ b/lazuardy_anfis/tests.py
@@ -11,12 +11,11 @@
 
 import os
 
+import anfis
+import membershipfunction
+import mfderivs
 import numpy as np
 
-import lazuardy_anfis.anfis
-import lazuardy_anfis.membershipfunction
-import lazuardy_anfis.mfderivs
-
 # define the training_set.txt file path
 training_set = os.getcwd() + "/lazuardy_anfis/training_set.txt"