diff --git a/bridge_former.py b/bridge_former.py
index 2109cb3..c25f1f5 100644
--- a/bridge_former.py
+++ b/bridge_former.py
@@ -84,7 +84,7 @@ class BridgeFormer(NNOperator):
 
         if self.model_name == "clip_initialized_model":
             text_data = tokenize(text).to(self.device)
-            text_features = self.model.encode_text(text_data["input_ids"])
+            text_features = self.model.encode_text(text_data)
         else:
             text_data = self.tokenizer(text, return_tensors='pt').to(self.device)
             text_features = self.model.compute_text(text_data)