diff --git a/src/memory/CudaMemory.cuh b/src/memory/CudaMemory.cuh
index 56b8d0eb4be24411a887f827ee1fea7ae631a749..6694637d784a9f577626d8193dfd96150901bd87 100644
--- a/src/memory/CudaMemory.cuh
+++ b/src/memory/CudaMemory.cuh
@@ -27,8 +27,12 @@
 #ifndef CUDA_MEMORY_CUH_
 #define CUDA_MEMORY_CUH_
 
+#include "config.h"
 #include "memory.hpp"
 #include <iostream>
+#ifdef SE_CLASS2
+#include "Memleak_check.hpp"
+#endif
 
 class CudaMemory : public memory
 {