diff --git a/tests/test_elements_worker/test_elements.py b/tests/test_elements_worker/test_elements.py
index 02c04ca34c83c7020ceec55217f61430f5fd822f..6e2f132bfc313d85567accea00f83dcb2e2eb5e2 100644
--- a/tests/test_elements_worker/test_elements.py
+++ b/tests/test_elements_worker/test_elements.py
@@ -976,17 +976,6 @@ def test_list_element_children_wrong_element(mock_elements_worker):
     )
 
 
-def test_list_element_children_wrong_best_class(mock_elements_worker):
-    elt = Element({"id": "12341234-1234-1234-1234-123412341234"})
-
-    with pytest.raises(AssertionError) as e:
-        mock_elements_worker.list_element_children(
-            element=elt,
-            best_class=1234,
-        )
-    assert str(e.value) == "best_class should be of type str or bool"
-
-
 def test_list_element_children_wrong_folder(mock_elements_worker):
     elt = Element({"id": "12341234-1234-1234-1234-123412341234"})
 
@@ -1031,15 +1020,15 @@ def test_list_element_children_wrong_type(mock_elements_worker):
     assert str(e.value) == "type should be of type str"
 
 
-def test_list_element_children_wrong_with_best_classes(mock_elements_worker):
+def test_list_element_children_wrong_with_classes(mock_elements_worker):
     elt = Element({"id": "12341234-1234-1234-1234-123412341234"})
 
     with pytest.raises(AssertionError) as e:
         mock_elements_worker.list_element_children(
             element=elt,
-            with_best_classes="not bool",
+            with_classes="not bool",
         )
-    assert str(e.value) == "with_best_classes should be of type bool"
+    assert str(e.value) == "with_classes should be of type bool"
 
 
 def test_list_element_children_wrong_with_corpus(mock_elements_worker):