diff --git a/db_fixture/test_data.py b/db_fixture/test_data.py
index 90b1542..0478838 100644
--- a/db_fixture/test_data.py
+++ b/db_fixture/test_data.py
@@ -1,10 +1,7 @@
 import sys
 sys.path.append('../db_fixture')
-try:
-    from mysql_db import DB
-except ImportError:
-    from .mysql_db import DB
 
+from db_fixture.mysql_db import DB
 
 
 # create data