diff --git a/src/include/kompute/Tensor.hpp b/src/include/kompute/Tensor.hpp
index 5d7524ee..a2bcd187 100644
--- a/src/include/kompute/Tensor.hpp
+++ b/src/include/kompute/Tensor.hpp
@@ -3,6 +3,7 @@
 
 #include "kompute/Core.hpp"
 #include "logger/Logger.hpp"
+#include <memory>
 #include <string>
 
 namespace kp {