diff --git a/src/Vector/cuda/vector_dist_cuda_func_test.cu b/src/Vector/cuda/vector_dist_cuda_func_test.cu
index 53bfc92353826354fc8ac7db1d5e2108528ba0e8..b0f330f6aa6c4bde9204b051ee6984fa8a5017e7 100644
--- a/src/Vector/cuda/vector_dist_cuda_func_test.cu
+++ b/src/Vector/cuda/vector_dist_cuda_func_test.cu
@@ -8,7 +8,7 @@
 #include "Vector/cuda/vector_dist_cuda_funcs.cuh"
 #include "Vector/util/vector_dist_funcs.hpp"
 #include "Decomposition/CartDecomposition.hpp"
-#include "util/cuda/scan_cuda.cuh"
+//#include "util/cuda/scan_cuda.cuh"
 #include "Vector/vector_dist.hpp"
 #include "util/cuda/scan_ofp.cuh"
 
diff --git a/src/Vector/vector_dist_kernel.hpp b/src/Vector/vector_dist_kernel.hpp
index 435515570da202555cb4e034bf2a1e2d16807b8e..c7cfb01c6ed554c2e170549abb03396f8eb96a5b 100644
--- a/src/Vector/vector_dist_kernel.hpp
+++ b/src/Vector/vector_dist_kernel.hpp
@@ -194,6 +194,11 @@ public:
 		return v_prp.template get<id>(vec_key.getKey());
 	}
 
+	__device__ __host__ int getOriginKey(int vec_key)
+	{
+		return vec_key;
+	}
+
 	/*! \brief Get the property of an element
 	 *
 	 * see the vector_dist iterator usage to get an element key