diff --git a/Gate_test.py b/Gate_test.py
index d79de8c..7ee97f0 100644
--- a/Gate_test.py
+++ b/Gate_test.py
@@ -1,4 +1,4 @@
-from QElephant.QuBit import *
+from QElephant import *
 import pytest
 
 ## MuBit
diff --git a/Matrix_test.py b/Matrix_test.py
index adaa4f6..af43c44 100644
--- a/Matrix_test.py
+++ b/Matrix_test.py
@@ -1,4 +1,4 @@
-from QElephant.Matrix import *
+from QElephant import *
 import pytest
 
 m = Matrix([[1, 2, 3], [4, 5, 6]])
diff --git a/QuBit_test.py b/QuBit_test.py
index 5a5f502..8d5f7c7 100644
--- a/QuBit_test.py
+++ b/QuBit_test.py
@@ -1,4 +1,4 @@
-from QElephant.QuBit import *
+from QElephant import *
 import pytest
 
 q = QuBit(0, 1)