diff --git a/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py b/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py
index 748212ce140aff5780b17893abbd1069efd97d0e..d9696e5a0b9ba8ae104c0157bf70d3d1be6596b9 100644
--- a/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py
+++ b/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py
@@ -98,7 +98,7 @@ def generate_ilastik_object_classifier(
         # delete existing classification weights
         if (k := f'ObjectExtraction/RegionFeatures/{lane:04d}') in h5.keys():
             del h5[k]
-        if (k := 'ObjectClassification/ClassificationForests') in h5.keys():
+        if (k := 'ObjectClassification/ClassifierForests') in h5.keys():
             del h5[k]
 
     return new_ilp
@@ -156,9 +156,7 @@ if __name__ == '__main__':
         label_names,
     )
 
-    from extensions.chaeo.examples.test_modify_object_classifier import modify_ilastik_object_classifier
-    auto_ilp = modify_ilastik_object_classifier(new_ilp)
-    # auto_ilp = new_ilp
+    auto_ilp = new_ilp
 
     def infer_and_compare(ilp, prefix, raw, mask, labels):
         mod = PatchStackObjectClassifier({'project_file': root / ilp})