1 files changed, 4 insertions, 3 deletions
diff --git a/main.py b/main.py
index aaa51da..8d03148 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,7 @@
 """Main entry point for project"""
+from yoyo import get_backend, read_migrations
 
+from gn3 import migrations
 from gn3.app import create_app
 
 app = create_app()
@@ -8,9 +10,8 @@ app = create_app()
 
 @app.cli.command()
 def apply_migrations():
-    from yoyo import get_backend, read_migrations
-    from gn3.migrations import apply_migrations
-    apply_migrations(
+    """Apply the dabasase migrations."""
+    migrations.apply_migrations(
         get_backend(f'sqlite:///{app.config["AUTH_DB"]}'),
         read_migrations(app.config["AUTH_MIGRATIONS"]))