diff --git a/kaldi_data_generator/main.py b/kaldi_data_generator/main.py
index 657e102998af3fb740a5a83674f9071a76eb5c65..41230882a414ae74a915850634f846c98d648164 100644
--- a/kaldi_data_generator/main.py
+++ b/kaldi_data_generator/main.py
@@ -172,13 +172,13 @@ class HTRDataGenerator:
         try:
             accepted_zones = []
             for elt in self.api_client.cached_paginate(
-                "ListElementChildren", id=page_id, with_best_classes=True
+                "ListElementChildren", id=page_id, with_classes=True
             ):
                 printed = True
-                for classification in elt["best_classes"]:
+                for classification in elt["classes"]:
                     if classification["ml_class"]["name"] == "handwritten":
                         printed = False
-                for classification in elt["best_classes"]:
+                for classification in elt["classes"]:
                     if classification["ml_class"]["name"] in self.accepted_classes:
                         if self.should_filter_printed:
                             if not printed:
@@ -336,7 +336,7 @@ class HTRDataGenerator:
         return {
             elem["id"]: [
                 best_class["ml_class"]["name"]
-                for best_class in elem["best_classes"]
+                for best_class in elem["classes"]
                 if best_class["state"] != "rejected"
             ]
             for elem in self.api_client.cached_paginate(
@@ -344,7 +344,7 @@ class HTRDataGenerator:
                 id=page_id,
                 recursive=True,
                 type=TEXT_LINE,
-                with_best_classes=True,
+                with_classes=True,
             )
         }