diff --git a/studio/backend/tests/test_diffusion_backend.py b/studio/backend/tests/test_diffusion_backend.py
index 947d9e7ae3..5aa8c9a71e 100644
--- a/studio/backend/tests/test_diffusion_backend.py
+++ b/studio/backend/tests/test_diffusion_backend.py
@@ -493,7 +493,9 @@ def test_generate_rejects_reference_on_unsupported_family(fake_runtime, tmp_path
     )
     with pytest.raises(ValueError, match = "Reference images are not supported"):
         backend.generate(
-            prompt = "x", steps = 4, init_image = _tiny_png_b64(),
+            prompt = "x",
+            steps = 4,
+            init_image = _tiny_png_b64(),
             reference_images = [_tiny_png_b64()],
         )