diff --git a/code/lib/model.py b/code/lib/model.py
index a2ac57b..9c442c0 100644
--- a/code/lib/model.py
+++ b/code/lib/model.py
@@ -112,7 +112,7 @@ def __define_criterion(self, class_weights, delta_var,
         # Discriminative Loss
         if self.use_instance_segmentation:
             self.criterion_discriminative = DiscriminativeLoss(
-                delta_var, delta_dist, norm, self.usegpu)
+                delta_var, delta_dist, norm, usegpu=self.usegpu)
             if self.usegpu:
                 self.criterion_discriminative = \
                     self.criterion_discriminative.cuda()