diff --git a/tests/test_training.py b/tests/test_training.py
index 3ce52d6c..fa3648ba 100644
--- a/tests/test_training.py
+++ b/tests/test_training.py
@@ -69,9 +69,7 @@ def test_training():
 
     assert workspace_latent.size() == (32, 16)
 
-    reconstructed_unimodal_latents = gw.decode(
-        workspace_latent, domains={"v", "a"}
-    )
+    reconstructed_unimodal_latents = gw.decode(workspace_latent, domains={"v", "a"})
 
     assert reconstructed_unimodal_latents.keys() == {"v", "a"}
     assert reconstructed_unimodal_latents["v"].size() == (32, 128)