diff --git a/surfsense_backend/app/schemas/base.py b/surfsense_backend/app/schemas/base.py
index 4bd7b2262..d357aabcb 100644
--- a/surfsense_backend/app/schemas/base.py
+++ b/surfsense_backend/app/schemas/base.py
@@ -1,8 +1,10 @@
 from datetime import datetime
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 class TimestampModel(BaseModel):
     created_at: datetime
+    model_config = ConfigDict(from_attributes=True)
 
 class IDModel(BaseModel):
-    id: int 
\ No newline at end of file
+    id: int
+    model_config = ConfigDict(from_attributes=True) 
\ No newline at end of file
diff --git a/surfsense_backend/app/schemas/documents.py b/surfsense_backend/app/schemas/documents.py
index 6a99fcde9..ad8bd9599 100644
--- a/surfsense_backend/app/schemas/documents.py
+++ b/surfsense_backend/app/schemas/documents.py
@@ -1,7 +1,5 @@
 from typing import List, Any
 from pydantic import BaseModel, ConfigDict
-from sqlalchemy import JSON
-from .base import IDModel, TimestampModel
 from app.db import DocumentType
 from datetime import datetime