diff --git a/test/nn/models/test_compile.py b/test/nn/models/test_compile.py
index dc22527c..79db1da5 100644
--- a/test/nn/models/test_compile.py
+++ b/test/nn/models/test_compile.py
@@ -54,7 +54,7 @@
             Trompt,
             dict(channels=8, num_prompts=2),
             None,
-            4,
+            3,
             id="Trompt",
         ),
         pytest.param(