diff --git a/cpu/src/neighbors.cpp b/cpu/src/neighbors.cpp
index 3a7556c..77578d7 100644
--- a/cpu/src/neighbors.cpp
+++ b/cpu/src/neighbors.cpp
@@ -3,6 +3,7 @@
 
 #include "neighbors.h"
 #include <chrono>
+#include <memory>
 #include <random>
 
 template <typename scalar_t>