diff --git a/include/LightGBM/cuda/cuda_column_data.hpp b/include/LightGBM/cuda/cuda_column_data.hpp
index 3935b54b771..964a3218a2b 100644
--- a/include/LightGBM/cuda/cuda_column_data.hpp
+++ b/include/LightGBM/cuda/cuda_column_data.hpp
@@ -101,7 +101,7 @@ class CUDAColumnData {
   std::vector<void*> GetDataByColumnPointers(const std::vector<std::unique_ptr<CUDAVector<void>>>& data_by_column) const {
     std::vector<void*> data_by_column_pointers(data_by_column.size(), nullptr);
     for (size_t i = 0; i < data_by_column.size(); ++i) {
-      data_by_column_pointers[i] = data_by_column[i].get();
+      data_by_column_pointers[i] = reinterpret_cast<void*>(data_by_column[i]->RawData());
     }
     return data_by_column_pointers;
   }