diff --git a/dpcpp/distributed/partition_kernels.dp.cpp b/dpcpp/distributed/partition_kernels.dp.cpp
index d3ad097acee..dfb91da4596 100644
--- a/dpcpp/distributed/partition_kernels.dp.cpp
+++ b/dpcpp/distributed/partition_kernels.dp.cpp
@@ -73,7 +73,7 @@ void setup_sizes_ids_permutation(
             permutation[i] = static_cast<GlobalIndexType>(i);
         },
         num_ranges, num_ranges, num_parts, range_offsets, range_parts,
-        range_sizes.get_data(), part_ids.get_data(), permutation.get_data());
+        range_sizes, part_ids, permutation);
 }
 
 
@@ -103,8 +103,8 @@ void compute_part_sizes_and_starting_indices(
                 prev_part == cur_part ? grouped_starting_indices[i - 1]
                                       : LocalIndexType{};
         },
-        num_ranges, range_sizes.get_const_data(), part_ids.get_const_data(),
-        permutation.get_const_data(), starting_indices, part_sizes);
+        num_ranges, range_sizes, part_ids, permutation, starting_indices,
+        part_sizes);
 }