diff --git a/db_revisions/env.py b/db_revisions/env.py
index 36d69c3..e441084 100644
--- a/db_revisions/env.py
+++ b/db_revisions/env.py
@@ -1,12 +1,10 @@
 import os
 from dotenv import load_dotenv
-
 from logging.config import fileConfig
-
 from sqlalchemy import engine_from_config
 from sqlalchemy import pool
-
 from alembic import context
+from src.models import Base
 
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.
@@ -20,7 +18,7 @@
 # add your model's MetaData object here
 # for 'autogenerate' support
 # from myapp import mymodel
-from src.models import Base
+
 target_metadata = Base.metadata
 
 # other values from the config, defined by the needs of env.py,
diff --git a/src/main.py b/src/main.py
index 4e82034..1111860 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,6 +1,5 @@
 import os
 import logging
-import sys
 import env  # noqa: F401
 from http import HTTPStatus
 from fastapi import FastAPI, HTTPException, Request
diff --git a/src/models.py b/src/models.py
index c1c6db8..4f0187e 100644
--- a/src/models.py
+++ b/src/models.py
@@ -1,5 +1,5 @@
 from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy import Column, DateTime, ForeignKey, Index, Integer, PrimaryKeyConstraint, String
+from sqlalchemy import Column, DateTime, ForeignKey, Index, PrimaryKeyConstraint, String
 from sqlalchemy.sql import func
 
 Base = declarative_base()