diff --git a/studio/backend/tests/test_diffusion_patch_backend.py b/studio/backend/tests/test_diffusion_patch_backend.py
index 3f038b8ccc..b175b0820c 100644
--- a/studio/backend/tests/test_diffusion_patch_backend.py
+++ b/studio/backend/tests/test_diffusion_patch_backend.py
@@ -36,7 +36,12 @@ def _torch(*, cuda = False, xpu = False):
     )
 
 
-def _modules(monkeypatch, *, torch = None, unsloth = False):
+def _modules(
+    monkeypatch,
+    *,
+    torch = None,
+    unsloth = False,
+):
     """Stub sys.modules so the gate sees a chosen torch / unsloth state."""
     mods = dict(sys.modules)
     mods.pop("unsloth", None)