Fix Flask-Migrate integration