diff --git a/tests/test_elements_worker/test_elements.py b/tests/test_elements_worker/test_elements.py
index 502a20cbf34a0c9b880a861ca8345d31df6b65c8..7eb714e046710ee28e9f96ea9d67641bb476d8db 100644
--- a/tests/test_elements_worker/test_elements.py
+++ b/tests/test_elements_worker/test_elements.py
@@ -350,11 +350,9 @@ def test_load_corpus_classes(responses, mock_elements_worker):
         ),
     ]
     assert mock_elements_worker.classes == {
-        "11111111-1111-1111-1111-111111111111": {
-            "good": "0000",
-            "average": "1111",
-            "bad": "2222",
-        }
+        "good": "0000",
+        "average": "1111",
+        "bad": "2222",
     }