diff --git a/ci/unit_tests/test_brain_image_synthesis_latent_diffusion_model.py b/ci/unit_tests/test_brain_image_synthesis_latent_diffusion_model.py
index 56d16cca..0543b403 100644
--- a/ci/unit_tests/test_brain_image_synthesis_latent_diffusion_model.py
+++ b/ci/unit_tests/test_brain_image_synthesis_latent_diffusion_model.py
@@ -25,7 +25,7 @@
         "gender": 1.0,
         "age": 0.7,
         "ventricular_vol": 0.7,
-        "brain_vol": 0.5
+        "brain_vol": 0.5,
     }
 ]