diff --git a/src/migrations/env.py b/src/migrations/env.py
index 41a6727..393a8aa 100644
--- a/src/migrations/env.py
+++ b/src/migrations/env.py
@@ -16,7 +16,7 @@ if config.config_file_name is not None:
 
 # add your model's MetaData object here
 # for 'autogenerate' support
-from models import Base
+from database import Base
 
 target_metadata = Base.metadata
 # target_metadata = None