diff --git a/setup.py b/setup.py
index ae30d8ede6f..41acb259164 100644
--- a/setup.py
+++ b/setup.py
@@ -132,6 +132,7 @@ def _setup_entry_points() -> Dict:
 
     # transformers integration
     for task in [
+        "masked_language_modeling",
         "question_answering",
         "text_classification",
         "token_classification",