diff --git a/src/cuda_backend.rs b/src/cuda_backend.rs
index 44e75575..5a80df08 100644
--- a/src/cuda_backend.rs
+++ b/src/cuda_backend.rs
@@ -1,6 +1,8 @@
 use crate::{CpuStorage, DType, Result, Shape};
 use cudarc::driver::CudaSlice;
 
+pub(crate) type Error = cudarc::driver::DriverError;
+
 #[derive(Debug, Clone)]
 pub struct CudaDevice(std::sync::Arc<cudarc::driver::CudaDevice>);
 
diff --git a/src/error.rs b/src/error.rs
index 1990b67a..94f5e50d 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -35,7 +35,7 @@ pub enum Error {
     },
 
     #[error(transparent)]
-    Cudarc(#[from] cudarc::driver::DriverError),
+    Cudarc(#[from] crate::cuda_backend::Error),
 }
 
 pub type Result<T> = std::result::Result<T, Error>;