diff --git a/include/cute/atom/copy_traits_sm90_im2col.hpp b/include/cute/atom/copy_traits_sm90_im2col.hpp
index d2cc0dce..06a19dbf 100644
--- a/include/cute/atom/copy_traits_sm90_im2col.hpp
+++ b/include/cute/atom/copy_traits_sm90_im2col.hpp
@@ -472,8 +472,8 @@ make_im2col_tma_copy_desc(
       tma_oob_fill);
 
   int driver_version = 0;
-  CUresult driver_version_result = cuDriverGetVersion(&driver_version);
-  assert(driver_version_result == CUDA_SUCCESS);
+  cudaError_t driver_version_err = cudaDriverGetVersion(&driver_version);
+  assert(driver_version_err == cudaSuccess);
   if (driver_version <= 13010) {
     if (cute::bits_to_bytes(
           cute::cosize(tensor_cwhdn.layout()) *
diff --git a/include/cute/atom/copy_traits_sm90_tma.hpp b/include/cute/atom/copy_traits_sm90_tma.hpp
index f69ec7b7..90dedd0b 100644
--- a/include/cute/atom/copy_traits_sm90_tma.hpp
+++ b/include/cute/atom/copy_traits_sm90_tma.hpp
@@ -1053,8 +1053,8 @@ make_tma_copy_desc(Tensor<GEngine,GLayout> const& gtensor,         // The origin
         tma_oobFill);
    
     int driver_version = 0;
-    CUresult driver_version_result = cuDriverGetVersion(&driver_version);
-    assert(driver_version_result == CUDA_SUCCESS);
+    cudaError_t driver_version_err = cudaDriverGetVersion(&driver_version);
+    assert(driver_version_err == cudaSuccess);
     if (driver_version <= 13010) {      
       if (cute::bits_to_bytes(
             cute::cosize(gtensor.layout()) *