diff --git a/dataset_builder/prepare_prompts_json.py b/dataset_builder/prepare_prompts_json.py
index e7337648a4..7faf66fa38 100644
--- a/dataset_builder/prepare_prompts_json.py
+++ b/dataset_builder/prepare_prompts_json.py
@@ -69,7 +69,7 @@ def main():
         help="Skips tests that fail to translate. By default, if a test fails to translate, the entire problem is skipped.",
     )
 
-    args.add_argument("--originals", type=str, default="../datasets/originals")
+    args.add_argument("--originals", type=str, default="../datasets/originals-with-cleaned-doctests")
 
     args = args.parse_args()