diff --git a/src/gpu/gpu_context.h b/src/gpu/gpu_context.h
index 718b07c..c7e1d80 100644
--- a/src/gpu/gpu_context.h
+++ b/src/gpu/gpu_context.h
@@ -52,6 +52,9 @@ public:
         cudaMalloc((void**)&crossmap_dev, (npixels+10) * (npixels+10) *sizeof(SUPPORTTYPE));
         cudaMalloc((void**)&intensities_avg_dev, npixels * npixels *sizeof(FLOAT));
 
+        cudaMalloc((void**)&densitysub_dev, npixels * npixels *sizeof(CUDA_COMPLEX));
+
+
         
         cudaMallocHost((void**)&intensities_host, npixels * npixels *sizeof(FLOAT));        
         cudaMallocHost((void**)&density_host, npixels * npixels *sizeof(CUDA_COMPLEX));
@@ -73,6 +76,7 @@ public:
         cudaMallocHost((void**)&densitybest_host, npixels * npixels *sizeof(CUDA_COMPLEX));
         cudaMallocHost((void**)&crossmap_host, (npixels+10) * (npixels+10) *sizeof(SUPPORTTYPE));
         cudaMallocHost((void**)&intensities_avg_host, npixels * npixels *sizeof(FLOAT));        
+        cudaMallocHost((void**)&densitysub_host, npixels * npixels *sizeof(CUDA_COMPLEX));