diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake
index 6aa6c6aceba2ac9b9c978c842ca3218fcacaf987..a6bf1a750c2b628b6c4df798bc2682fa3eef75f5 100644
--- a/cmake/cuda.cmake
+++ b/cmake/cuda.cmake
@@ -12,7 +12,7 @@ if(ENABLE_CUDA)
 
   set(CUDA_HOST_COMPILER ${CMAKE_CXX_COMPILER})
   set(CUDA_PROPAGATE_HOST_FLAGS OFF)
-
+  add_definitions(-DENABLE_CUDA)
   # Pass options to NVCC ( -ccbin /path  --compiler-options -lfftw3f --compiler-options -lm --verbose)
   list(APPEND CUDA_NVCC_FLAGS -DENABLE_CUDA --std c++11 -Wno-deprecated-gpu-targets)
   list(APPEND CUDA_NVCC_FLAGS_DEBUG --debug; --device-debug; --generate-line-info -Xcompiler "-Werror")
diff --git a/psrdada_cpp/src/dada_read_client.cpp b/psrdada_cpp/src/dada_read_client.cpp
index e25ebca4d7df997b1f26de2c13b426e06f533ede..3fb504f25a99f654f081a6c07ae1c73c769dd545 100644
--- a/psrdada_cpp/src/dada_read_client.cpp
+++ b/psrdada_cpp/src/dada_read_client.cpp
@@ -10,7 +10,7 @@ namespace psrdada_cpp {
     {
 
         BOOST_LOG_TRIVIAL(debug) << this->id() << "Pinning dada buffers for CUDA memcpy";
-        dada_cuda_dbregister(_hdu);
+        cuda_register_memory();
         lock();
     }