diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 39dbe80..6e6b4e1 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -74,9 +74,9 @@ class Model(pl.LightningModule):
     def loss(self, loss):
 
         if isinstance(loss, str):
-            losses = [loss]
+            loss = [loss]
 
-        self._loss = Avergeloss(losses)
+        self._loss = Avergeloss(loss)
 
     @property
     def metric(self):