diff --git a/tests/test_migrate.py b/tests/test_migrate.py
index b33a94d..2ca6826 100644
--- a/tests/test_migrate.py
+++ b/tests/test_migrate.py
@@ -1,6 +1,5 @@
 import os
 import shutil
-from exceptions import OSError
 import unittest
 
 class TestMigrate(unittest.TestCase):
@@ -30,7 +29,7 @@ def tearDown(self):
             pass
 
     def test_migrate_upgrade(self):
-        from app import db, User
+        from .app import db, User
         db.session.add(User(name = 'test'))
         db.session.commit()
 
diff --git a/tests/test_migrate_custom_directory.py b/tests/test_migrate_custom_directory.py
index 7451550..88cb3b7 100644
--- a/tests/test_migrate_custom_directory.py
+++ b/tests/test_migrate_custom_directory.py
@@ -1,6 +1,5 @@
 import os
 import shutil
-from exceptions import OSError
 import unittest
 
 class TestMigrate(unittest.TestCase):
@@ -30,7 +29,7 @@ def tearDown(self):
             pass
 
     def test_migrate_upgrade(self):
-        from app2 import db, User
+        from .app2 import db, User
         db.session.add(User(name = 'test'))
         db.session.commit()