diff --git a/dreambooth_runpod_joepenna.ipynb b/dreambooth_runpod_joepenna.ipynb
index 2c5f7d8..1674d5f 100644
--- a/dreambooth_runpod_joepenna.ipynb
+++ b/dreambooth_runpod_joepenna.ipynb
@@ -394,7 +394,7 @@
     " --gpus 0, \\\n",
     " --data_root \"/workspace/Dreambooth-Stable-Diffusion/training_samples\" \\\n",
     " --max_training_steps {max_training_steps} \\\n",
-    " --class_word class_word \\\n",
+    " --class_word {class_word} \\\n",
     " --no-test"
    ]
   },