diff --git a/agent/application/states/database.py b/agent/application/states/database.py
index 78756b8..745f217 100644
--- a/agent/application/states/database.py
+++ b/agent/application/states/database.py
@@ -9,9 +9,10 @@ from random import choices
 from pydantic_ai import ModelMessage, ModelMessagesTypeAdapter, RunUsage
 from pydantic_ai._uuid import uuid7
 import reflex as rx
-from sqlmodel import Field, JSON, SQLModel, select, update
+from sqlmodel import Field, SQLModel, select, update
 
 from application.states.models import (
+    Approval,
     Conversation,
     Message,
     MessageType,
@@ -22,7 +23,6 @@ from application.states.models import (
     usage_validate_json,
     workflow_dump_json,
     workflow_validate_json,
-    Approval,
 )