diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 00efc6851a82fc3cf4ac25403bd15d9320a8df5d..d8c3602cc13c6ebd3a19acc4c73ea738272da723 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -1576,7 +1576,7 @@ class DomainAdaptationTrainer(ClassificationNetworkTrainer):
         target = iter(self.trg_train_dl)
 
         # increase domain adaptation weight with increasing epochs
-        uda_lambda = self.uda_lambda * ((epoch + 1) / self.epochs)
+        # uda_lambda = self.uda_lambda * ((epoch + 1) / self.epochs)
 
         # iterate over the number of samples
         for batch, (src_input, src_label) in enumerate(self.src_train_dl):
@@ -1610,7 +1610,7 @@ class DomainAdaptationTrainer(ClassificationNetworkTrainer):
             uda_loss = self.uda_loss_function(src_feature, trg_feature)
 
             # total loss
-            tot_loss = cla_loss + uda_lambda * uda_loss
+            tot_loss = cla_loss + self.uda_lambda * uda_loss
 
             # compute the gradients of the loss function w.r.t.
             # the network weights