diff --git a/courses/machine_learning/deepdive/09_sequence/poetry.ipynb b/courses/machine_learning/deepdive/09_sequence/poetry.ipynb
index b2976744a1..2d4b2704cd 100644
--- a/courses/machine_learning/deepdive/09_sequence/poetry.ipynb
+++ b/courses/machine_learning/deepdive/09_sequence/poetry.ipynb
@@ -631,7 +631,7 @@
     "t2t-trainer \\\n",
     "  --data_dir=gs://${BUCKET}/poetry/subset \\\n",
     "  --t2t_usr_dir=./poetry/trainer \\\n",
-    "  --problems=$PROBLEM \\\n",
+    "  --problem=$PROBLEM \\\n",
     "  --model=transformer \\\n",
     "  --hparams_set=transformer_poetry \\\n",
     "  --output_dir=$OUTDIR --job-dir=$OUTDIR --train_steps=10"
@@ -666,7 +666,7 @@
     "echo \"Y\" | t2t-trainer \\\n",
     "  --data_dir=gs://${BUCKET}/poetry/subset \\\n",
     "  --t2t_usr_dir=./poetry/trainer \\\n",
-    "  --problems=$PROBLEM \\\n",
+    "  --problem=$PROBLEM \\\n",
     "  --model=transformer \\\n",
     "  --hparams_set=transformer_poetry \\\n",
     "  --output_dir=$OUTDIR \\\n",