diff --git a/torchbenchmark/models/clip/__init__.py b/torchbenchmark/models/clip/__init__.py
index 9f6a35f765..7ad040377e 100644
--- a/torchbenchmark/models/clip/__init__.py
+++ b/torchbenchmark/models/clip/__init__.py
@@ -40,7 +40,7 @@ def __init__(self, test, device, jit=False, batch_size=1, extra_args=[]):
         # Create optimizer
         self.loss_fn = ContrastiveLossWithTemperature()
         self.optimizer = torch.optim.AdamW(
-            list(self.model.parameters()) + list(loss_fn.parameters()),
+            list(self.model.parameters()) + list(self.loss_fn.parameters()),
             lr=5.0e-4,
             weight_decay=1.0e-4,
             eps=1.0e-6,