diff --git a/stable-diffusion-finetuning/pokemon_finetune.ipynb b/stable-diffusion-finetuning/pokemon_finetune.ipynb
index 79aa765..dd3612c 100644
--- a/stable-diffusion-finetuning/pokemon_finetune.ipynb
+++ b/stable-diffusion-finetuning/pokemon_finetune.ipynb
@@ -142,7 +142,7 @@
         "    --finetune_from \"$ckpt_path\" \\\n",
         "    data.params.batch_size=\"$BATCH_SIZE\" \\\n",
         "    lightning.trainer.accumulate_grad_batches=\"$ACCUMULATE_BATCHES\" \\\n",
-        "    data.params.validation.params.n_gpus=\"$NUM_GPUS\" \\\n",
+        "    data.params.validation.params.n_gpus=\"$N_GPUS\" \\\n",
         ")"
       ]
     },