diff --git a/codeflash/models/models.py b/codeflash/models/models.py
index 9234a3256..a50c8143b 100644
--- a/codeflash/models/models.py
+++ b/codeflash/models/models.py
@@ -210,13 +210,7 @@ def to_string(self) -> str:
         )
 
     def to_dict(self) -> dict[str, Any]:
-        return {
-            "benchmark_name": self.benchmark_name,
-            "test_function": self.test_function,
-            "original_timing": self.original_timing,
-            "expected_new_timing": self.expected_new_timing,
-            "speedup_percent": self.speedup_percent,
-        }
+        return self.__dict__.copy()
 
 
 @dataclass