diff --git a/migrations/versions/26a14ac7025d_.py b/migrations/versions/26a14ac7025d_.py
index 506c90a..73a0647 100644
--- a/migrations/versions/26a14ac7025d_.py
+++ b/migrations/versions/26a14ac7025d_.py
@@ -51,7 +51,8 @@ def upgrade() -> None:
         sa.Column('total_tokens', sa.Integer(), nullable=False),
         sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=True),
         sa.ForeignKeyConstraint(['agent_id'], ['evo_core_agents.id'], ondelete='CASCADE'),
-        sa.PrimaryKeyConstraint('id')
+        sa.PrimaryKeyConstraint('id'),
+        if_not_exists=True,
     )