5 files changed, 7 insertions, 7 deletions
diff --git a/tests/conftest.py b/tests/conftest.py
index 6a2d0a6..505c83d 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -10,8 +10,8 @@ import os
 import tempfile
 
 import pytest
-from flaskr import create_app
-from flaskr.db import get_db, init_db
+from ustayml import create_app
+from ustayml.db import get_db, init_db
 
 with open(os.path.join(os.path.dirname(__file__), 'data.sql'), 'rb') as f:
     _data_sql = f.read().decode('utf8')
diff --git a/tests/test_auth.py b/tests/test_auth.py
index 401b61f..6684f43 100644
--- a/tests/test_auth.py
+++ b/tests/test_auth.py
@@ -1,6 +1,6 @@
 import pytest
 from flask import g, session
-from flaskr.db import get_db
+from ustayml.db import get_db
 
 
 def test_register(client, app):
diff --git a/tests/test_blog.py b/tests/test_blog.py
index 75f67a5..06b6437 100644
--- a/tests/test_blog.py
+++ b/tests/test_blog.py
@@ -1,5 +1,5 @@
 import pytest
-from flaskr.db import get_db
+from ustayml.db import get_db
 
 
 def test_index(client, auth):
diff --git a/tests/test_db.py b/tests/test_db.py
index 6ac4635..4e2f103 100644
--- a/tests/test_db.py
+++ b/tests/test_db.py
@@ -1,7 +1,7 @@
 import sqlite3
 
 import pytest
-from flaskr.db import get_db
+from ustayml.db import get_db
 
 
 def test_get_close_db(app):
@@ -27,7 +27,7 @@ def test_init_db_command(runner, monkeypatch):
     def fake_init_db():
         Recorder.called = True
 
-    monkeypatch.setattr('flaskr.db.init_db', fake_init_db)
+    monkeypatch.setattr('ustayml.db.init_db', fake_init_db)
     result = runner.invoke(args=['init-db'])
     assert 'Initialized' in result.output
     assert Recorder.called
 \ No newline at end of file
diff --git a/tests/test_factory.py b/tests/test_factory.py
index 024c19a..edc973c 100644
--- a/tests/test_factory.py
+++ b/tests/test_factory.py
@@ -1,4 +1,4 @@
-from flaskr import create_app
+from ustayml import create_app
 
 
 def test_config():