diff --git a/custom_utils/sql.py b/custom_utils/sql.py
index <HASH>..<HASH> 100644
--- a/custom_utils/sql.py
+++ b/custom_utils/sql.py
@@ -1,5 +1,6 @@
 import sqlalchemy
-from sqlalchemy import Column, ForeignKey, UniqueConstraint, Integer, String, DateTime, create_engine, update
+from sqlalchemy import Column, ForeignKey, UniqueConstraint, PrimaryKeyConstraint
+from sqlalchemy import create_engine, update, Integer, String, DateTime
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import relationship, sessionmaker
Added PrimaryKeyConstraint import from sqlalchemy