diff --git a/worker/orca-agent/requirements.txt b/worker/orca-agent/requirements.txt
index ddfc110..a5895dd 100644
--- a/worker/orca-agent/requirements.txt
+++ b/worker/orca-agent/requirements.txt
@@ -16,5 +16,5 @@ tenacity>=9.0.0
 sqlmodel>=0.0.22
 openai>=0.28.0
 colorama>=0.4.6
-prometheus-swarm>=0.1.0
+prometheus-swarm>=0.1.2
 prometheus-test>=0.1.1
diff --git a/worker/orca-agent/src/database/models.py b/worker/orca-agent/src/database/models.py
index c361904..7a83ad5 100644
--- a/worker/orca-agent/src/database/models.py
+++ b/worker/orca-agent/src/database/models.py
@@ -1,11 +1,9 @@
 """Database models."""
 
-from datetime import datetime
-from typing import Optional, List
-from sqlmodel import SQLModel, Field, Relationship
+from typing import Optional
+from sqlmodel import SQLModel, Field
 from sqlalchemy import JSON
 from sqlalchemy import Column
-from prometheus_swarm.database.models import Conversation, Message, Log
 
 
 class Submission(SQLModel, table=True):
@@ -19,4 +17,4 @@ class Submission(SQLModel, table=True):
     repo_urls: Optional[dict] = Field(
         default=None, sa_column=Column(JSON)
     )  # Store as JSON type
-    repo_url: Optional[str] = None
\ No newline at end of file
+    repo_url: Optional[str] = None
diff --git a/worker/orca-agent/src/server/routes/submission.py b/worker/orca-agent/src/server/routes/submission.py
index a4efa34..dcaeb66 100644
--- a/worker/orca-agent/src/server/routes/submission.py
+++ b/worker/orca-agent/src/server/routes/submission.py
@@ -1,6 +1,6 @@
 from flask import Blueprint, jsonify
 from prometheus_swarm.database import get_db
-from src.dababase.models import Submission
+from src.database.models import Submission
 import logging
 import os