diff --git a/tests/unit/test_fmha_6warp_multirow.cu b/tests/unit/test_fmha_6warp_multirow.cu
index 93a5dfcc..d9f14f81 100644
--- a/tests/unit/test_fmha_6warp_multirow.cu
+++ b/tests/unit/test_fmha_6warp_multirow.cu
@@ -113,7 +113,7 @@ static int test_single_T(int T, int n_h = 1, int batch = 1) {
     fmha_6warp_multirow_kernel<HD><<<grid, 192, smem>>>(params);
 
     cudaError_t err = cudaDeviceSynchronize();
-    if (err != cudaErrorCudaSuccess) {
+    if (err != cudaSuccess) {
         printf("  CUDA ERROR: %s\n", cudaGetErrorString(err));
         cudaFree(d_q); cudaFree(d_k); cudaFree(d_v); cudaFree(d_o); cudaFree(d_lse);
         free(h_q); free(h_k); free(h_v); free(h_o); free(h_lse);