diff --git a/tests/test_nacl_loss.py b/tests/test_nacl_loss.py
index 608da95301..a587509051 100644
--- a/tests/test_nacl_loss.py
+++ b/tests/test_nacl_loss.py
@@ -46,22 +46,8 @@
 targets = (torch.tensor([[[1, 1, 1, 1], [1, 1, 1, 0], [0, 0, 0, 0], [0, 0, 0, 0]]]),)
 
 TEST_CASES = [
-    [
-        {"classes": 3, "dim": 2},
-        {
-            "inputs": inputs,
-            "targets": targets,
-        },
-        1.1820,
-    ],
-    [
-        {"classes": 3, "dim": 2, "kernel_ops": "gaussian"},
-        {
-            "inputs": inputs,
-            "targets": targets,
-        },
-        1.1850,
-    ],
+    [{"classes": 3, "dim": 2}, {"inputs": inputs, "targets": targets}, 1.1820],
+    [{"classes": 3, "dim": 2, "kernel_ops": "gaussian"}, {"inputs": inputs, "targets": targets}, 1.1850],
     [
         {"classes": 3, "dim": 3, "kernel_ops": "gaussian"},
         {