diff --git a/docs/sections/pipeline_samples/papers/ultrafeedback.md b/docs/sections/pipeline_samples/papers/ultrafeedback.md
index a245d121c1..704309e263 100644
--- a/docs/sections/pipeline_samples/papers/ultrafeedback.md
+++ b/docs/sections/pipeline_samples/papers/ultrafeedback.md
@@ -210,7 +210,7 @@ distiset = pipeline.run(
                 }
             },
         },
-        ultrafeedback_openai.name: {
+        ultrafeedback.name: {
             "llm": {
                 "generation_kwargs": {
                     "max_new_tokens": 2048,