diff --git a/Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg.cpp b/Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg.cpp
index eec3f82891..d5733f3af3 100644
--- a/Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg.cpp
+++ b/Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg.cpp
@@ -380,7 +380,7 @@ int run_sparse_pcg_example(const sycl::device &dev)
         // setup optimizations and properties we know about A matrix
         oneapi::mkl::sparse::init_matrix_handle(&A);
 
-        auto ev_set = oneapi::mkl::sparse::set_csr_data(q, A, n, n,
+        auto ev_set = oneapi::mkl::sparse::set_csr_data(q, A, n, n, nnz,
                 oneapi::mkl::index_base::zero, ia_d, ja_d, a_d, {});
 
         oneapi::mkl::sparse::set_matrix_property(A, oneapi::mkl::sparse::property::symmetric);
diff --git a/Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg2.cpp b/Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg2.cpp
index adb2360ac8..3fd341d0e9 100644
--- a/Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg2.cpp
+++ b/Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg2.cpp
@@ -453,7 +453,7 @@ int run_sparse_pcg_example(const sycl::device &dev)
         // setup optimizations and properties we know about A matrix
         oneapi::mkl::sparse::init_matrix_handle(&A);
 
-        auto ev_set = oneapi::mkl::sparse::set_csr_data(q, A, n, n,
+        auto ev_set = oneapi::mkl::sparse::set_csr_data(q, A, n, n, nnz,
                 oneapi::mkl::index_base::zero, ia_d, ja_d, a_d, {});
 
         oneapi::mkl::sparse::set_matrix_property(A, oneapi::mkl::sparse::property::symmetric);