diff --git a/tensor2tensor/notebooks/Transformer_translate.ipynb b/tensor2tensor/notebooks/Transformer_translate.ipynb
index 4e9925498..4f26a943c 100644
--- a/tensor2tensor/notebooks/Transformer_translate.ipynb
+++ b/tensor2tensor/notebooks/Transformer_translate.ipynb
@@ -351,7 +351,7 @@
         "  --schedule=$schedule\\\n",
         "  --output_dir=$TRAIN_DIR \\\n",
         "  --train_steps=$train_steps \\\n",
-        "  --worker-gpu=1 \\ \n",
+        "  --worker_gpu=1 \\ \n",
         "  --eval_steps=$eval_steps "
       ],
       "execution_count": 0,
@@ -364,7 +364,7 @@
         "colab_type": "text"
       },
       "source": [
-        "  --worker-gpu = 1, for train on 1 gpu (facultative).\n",
+        "  --worker_gpu = 1, for train on 1 gpu (facultative).\n",
         "\n",
         "---\n",
         "\n",