diff --git a/agent/models.py b/agent/models.py
index 81a2d10..0db1fef 100644
--- a/agent/models.py
+++ b/agent/models.py
@@ -1,5 +1,5 @@
 from pydantic import BaseModel, Field, validator
-from typing import List
+from typing import List, Any
 from .constants import MIN_SCORE, MAX_SCORE
 
 
@@ -30,7 +30,7 @@ class FinalResult(BaseModel):
     best_score: float = Field(description="The best score for the final tweet")
     iterations_run: int = Field(description="The number of iterations run")
     early_stopped: bool = Field(description="Whether the optimization early stopped")
-    scores_history: List[List[float]] = Field(description="The history of scores")
+    scores_history: List[Any] = Field(description="The history of scores")
     improvement_count: int = Field(description="The number of improvements found")