diff --git a/docs/source/using-the-python-api.mdx b/docs/source/using-the-python-api.mdx
index e2ee0adae..9ad7f3c90 100644
--- a/docs/source/using-the-python-api.mdx
+++ b/docs/source/using-the-python-api.mdx
@@ -34,6 +34,7 @@ def main():
     pipeline_params = PipelineParameters(
         launcher_type=ParallelismManager.ACCELERATE,
         env_config=EnvConfig(cache_dir="tmp/"),
+        custom_task_directory=None, # if using a custom task
         # Remove the 2 parameters below once your configuration is tested
         override_batch_size=1,
         max_samples=10
@@ -52,7 +53,6 @@ def main():
         pipeline_parameters=pipeline_params,
         evaluation_tracker=evaluation_tracker,
         model_config=model_config,
-        custom_task_directory=None, # if using a custom task
     )
 
     pipeline.evaluate()