diff --git a/vision_agent/utils/sim.py b/vision_agent/utils/sim.py
index 5c89f700..8bb9b205 100644
--- a/vision_agent/utils/sim.py
+++ b/vision_agent/utils/sim.py
@@ -125,7 +125,9 @@ def __init__(
             raise ValueError("key is required if no column 'embs' is present.")
 
         if sim_key is not None:
-            self.df["embs"] = self.df[sim_key].apply(lambda x: get_embedding(client, x))
+            self.df["embs"] = self.df[sim_key].apply(
+                lambda x: get_embedding(self.emb_call, x)
+            )
 
 
 class OllamaSim(Sim):