diff --git a/migrations/env.py b/migrations/env.py
index 04f6113..16c5051 100644
--- a/migrations/env.py
+++ b/migrations/env.py
@@ -3,7 +3,7 @@
 from alembic import context
 from sqlalchemy import engine_from_config, pool
 
-from .models.base import BaseDbModel
+from models import *
 from rating_api.settings import get_settings