diff --git a/vcache/vcache_policy/strategies/benchmark_verified_global.py b/vcache/vcache_policy/strategies/benchmark_verified_global.py
index 8fa8842..a6488d7 100644
--- a/vcache/vcache_policy/strategies/benchmark_verified_global.py
+++ b/vcache/vcache_policy/strategies/benchmark_verified_global.py
@@ -215,9 +215,9 @@ def update_metadata(
             metadata: The metadata of the embedding.
         """
         if is_correct:
-            self.global_observations.append((round(similarity_score, 3), 1))
+            self.global_observations.append((round(similarity_score, 3).item(), 1))
         else:
-            self.global_observations.append((round(similarity_score, 3), 0))
+            self.global_observations.append((round(similarity_score, 3).item(), 0))
 
     def select_action(
         self, similarity_score: float, metadata: EmbeddingMetadataObj
diff --git a/vcache/vcache_policy/strategies/verified.py b/vcache/vcache_policy/strategies/verified.py
index f37289c..d27901a 100644
--- a/vcache/vcache_policy/strategies/verified.py
+++ b/vcache/vcache_policy/strategies/verified.py
@@ -484,9 +484,9 @@ def add_observation_to_metadata(
             metadata: EmbeddingMetadataObj - The metadata of the embedding
         """
         if is_correct:
-            metadata.observations.append((round(similarity_score, 3), 1))
+            metadata.observations.append((round(similarity_score, 3).item(), 1))
         else:
-            metadata.observations.append((round(similarity_score, 3), 0))
+            metadata.observations.append((round(similarity_score, 3).item(), 0))
 
     def select_action(
         self, similarity_score: float, metadata: EmbeddingMetadataObj