diff --git a/train.py b/train.py
index 2c9f0ecd..e82dbf23 100755
--- a/train.py
+++ b/train.py
@@ -229,7 +229,7 @@ def get_ganloss_weight(step):
                     
                     # Calculate adapted weight based on audio-visual similarity
                     if cfg.use_adapted_weight:
-                        vision_embed_gt = syncnet.get_vision_embed(gt_frames)
+                        vision_embed_gt = syncnet.get_image_embed(gt_frames)
                         image_audio_sim_gt = F.cosine_similarity(
                             audio_embed, 
                             vision_embed_gt,