diff --git a/studio/backend/core/training/trainer.py b/studio/backend/core/training/trainer.py
index a9d9da3978..b20d55aff8 100644
--- a/studio/backend/core/training/trainer.py
+++ b/studio/backend/core/training/trainer.py
@@ -3062,8 +3062,7 @@ class UnslothTrainer:
                     # large documents still works.
                     vision_image_size = training_args.get("vision_image_size")
                     deepseek_image_size = (
-                        640 if vision_image_size is None
-                        else int(vision_image_size)
+                        640 if vision_image_size is None else int(vision_image_size)
                     )
                     if vision_image_size is not None:
                         logger.info(