diff --git a/tests/unit/test_umma_qk.cu b/tests/unit/test_umma_qk.cu
index 775f4fc2..878d4ed4 100644
--- a/tests/unit/test_umma_qk.cu
+++ b/tests/unit/test_umma_qk.cu
@@ -180,7 +180,7 @@ int main() {
             h_s_out[r*8+0], h_s_out[r*8+1], h_s_out[r*8+2], h_s_out[r*8+3]);
     }
 
-    printf("Test %s\n", rel_err < 0.01f ? "PASSED" : "FAILED");
+    printf("Test %s\n", row0_rel < 0.001f ? "PASSED" : "FAILED");
     cudaFree(d_q); cudaFree(d_k); cudaFree(d_s_out); cudaFree(d_s_scalar);
     free(h_q); free(h_k); free(h_s_out); free(h_s_scalar);
     return 0;