diff --git a/arkindex/training/api.py b/arkindex/training/api.py
index f0b71f16bcd8564738f18a24be8b37903dd8e35f..46989ec6d6e49f7870777c95a6cce02e08c0ba28 100644
--- a/arkindex/training/api.py
+++ b/arkindex/training/api.py
@@ -790,6 +790,14 @@ class DatasetElements(CorpusACLMixin, ListCreateAPIView):
 @extend_schema_view(
     delete=extend_schema(
         operation_id="DestroyDatasetElement",
+        parameters=[
+            OpenApiParameter(
+                "set",
+                type=str,
+                description="Name of the set from which to remove the element.",
+                required=True,
+            )
+        ],
         tags=["datasets"]
     )
 )