diff --git a/Learning/ObjectBased.py b/Learning/ObjectBased.py
index 893455d8c58557e65a5427ddfee5db1fc47d62eb..f1dd96c4a70a74ad43ebe74ef4af33bd124e0af3 100644
--- a/Learning/ObjectBased.py
+++ b/Learning/ObjectBased.py
@@ -9,8 +9,10 @@ from sklearn.ensemble import RandomForestClassifier
 from sklearn.metrics import confusion_matrix, accuracy_score, cohen_kappa_score, precision_recall_fscore_support
 
 class ObjectBasedClassifier:
-    def __init__(self, object_layer, reference_data, time_series_patterns, user_feature_list, ref_class_field='class'):
-        self.obia_base = OBIABase(object_layer, ref_data = reference_data, ref_class_field=ref_class_field)
+    def __init__(self, object_layer, reference_data, time_series_patterns, user_feature_list, ref_class_field='class',
+                 ref_id_field='id'):
+        self.obia_base = OBIABase(object_layer, ref_data=reference_data, ref_class_field=ref_class_field,
+                                  ref_id_field=ref_id_field)
         for ptrn in time_series_patterns:
             lst = sorted(glob.glob(ptrn))
             self.obia_base.add_raster_time_series_for_stats(lst)