diff --git a/harvester/db/models.py b/harvester/db/models.py
index d6b834a..860e693 100644
--- a/harvester/db/models.py
+++ b/harvester/db/models.py
@@ -1,9 +1,5 @@
-# harvester/db/models.py
-
 from sqlalchemy import Column, Integer, String, DateTime
-from sqlalchemy.ext.declarative import declarative_base
-
-Base = declarative_base()
+from .schema import Base
 
 
 class Proxy(Base):
@@ -21,5 +17,5 @@ class Proxy(Base):
     def __repr__(self):
         return (
             f'Proxy(id={self.id}, host={self.host}, port={self.port}, '
-            f'egress={self.egress}, created_at={self.created_at})'
+            f'egress_ip={self.egress_ip}, date_added={self.created_at})'
         )