diff --git a/tests/unit/test_qk_softmax.cu b/tests/unit/test_qk_softmax.cu
index 8bd009fe..aacd2605 100644
--- a/tests/unit/test_qk_softmax.cu
+++ b/tests/unit/test_qk_softmax.cu
@@ -172,7 +172,7 @@ test_qk_softmax_kernel(
 
 int main() {
     printf("QK+Softmax Test (HD=%d, SK=%d)\n", (int)HD, (int)SK);
-    const int T = 4;
+    const int T = 1;
     const float SCALE = 1.0f / sqrtf((float)HD);
 
     bf16_t* h_q = (bf16_t*)calloc(T * HD, sizeof(bf16_t));