diff --git a/xpu/linear_algebra.hpp b/xpu/linear_algebra.hpp
index 87426db..273113f 100644
--- a/xpu/linear_algebra.hpp
+++ b/xpu/linear_algebra.hpp
@@ -555,7 +555,7 @@ class cholesky_factorization {
 #if defined(XPU_CUDA)
     detail::cusolver_potrf(
       handle_, order_, stride_,
-      matrix, workspace_.data(), workspace_.size(), info_.data()
+      matrix, workspace_.data(), workspace_.count(), info_.data()
     );
 
     auto info{0};