diff --git a/src/sparse/impl/KokkosSparse_spgemm_impl.hpp b/src/sparse/impl/KokkosSparse_spgemm_impl.hpp
index 19e576eb9d..06a3153ad9 100644
--- a/src/sparse/impl/KokkosSparse_spgemm_impl.hpp
+++ b/src/sparse/impl/KokkosSparse_spgemm_impl.hpp
@@ -808,7 +808,7 @@ class KokkosSPGEMM{
       num_chunks = (free_byte / 2) / chunk_bytes;
     }
     //then take the largest power of 2 smaller than that
-    nnz_lno_t po2_num_chunks = 1;
+    size_t po2_num_chunks = 1;
     while (po2_num_chunks * 2 < num_chunks) {
       po2_num_chunks *= 2;
     }