diff --git a/docker-compose.yml b/docker-compose.yml
index 059c028..85410a3 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,6 +11,8 @@ services:
       - "./lb:/usr/local/etc/haproxy"
     depends_on:
       - "certbot"
+      - "flask"
+      - "osmtile"
     restart: always
     networks:
       - internal
diff --git a/flask/manage.py b/flask/manage.py
index 110ed75..85c9a91 100644
--- a/flask/manage.py
+++ b/flask/manage.py
@@ -1,18 +1,16 @@
 #!/usr/bin/env python
 
 import os
-import subprocess, shlex
-from forest import app, db, User
 
 from flask.cli import FlaskGroup
+from forest import app, db, User
+
 from flask_migrate import Migrate
-from flask_migrate import MigrateCommand
 
 cli = FlaskGroup(app)
-migrate = Migrate(app, db)
-
-cli.add_command('db', MigrateCommand)
 
+migrate = Migrate()
+migrate.init_app(app, db)
 
 @cli.command("create_db")
 def create_db():