diff --git a/models/unet_lightning.py b/models/unet_lightning.py
index 38e6d47..dc2cd7d 100644
--- a/models/unet_lightning.py
+++ b/models/unet_lightning.py
@@ -239,7 +239,7 @@ class DiceLoss(nn.Module):
     def __init__(self, weight=None, size_average=True):
         super(DiceLoss, self).__init__()
 
-    def forward(self, targets, inputs, smooth=1):
+    def forward(self, inputs, targets, smooth=1):
         
         #comment out if your model contains a sigmoid or equivalent activation layer
         inputs = F.sigmoid(inputs)