diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 1a2ef97465beb44f052d3b1a9354924706e192d4..00efc6851a82fc3cf4ac25403bd15d9320a8df5d 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -1668,7 +1668,7 @@ class DomainAdaptationTrainer(ClassificationNetworkTrainer):
                 'trg_train_dl': self.trg_train_dl,
                 'trg_valid_dl': self.trg_valid_dl,
                 'trg_test_dl': self.trg_test_dl,
-                'uda': self.uda,
+                'supervised': self.supervised,
                 'uda_pos': self.uda_pos,
                 'uda_lambda': self.uda_lambda}