diff --git a/studio/frontend/src/features/images/diffusion-train-dialog.tsx b/studio/frontend/src/features/images/diffusion-train-dialog.tsx
index cf8fdb4b90..85fe0dd99b 100644
--- a/studio/frontend/src/features/images/diffusion-train-dialog.tsx
+++ b/studio/frontend/src/features/images/diffusion-train-dialog.tsx
@@ -166,8 +166,9 @@ export function DiffusionTrainDialog({