diff --git a/include/octree.h b/include/octree.h
index 723c911..86bb29c 100644
--- a/include/octree.h
+++ b/include/octree.h
@@ -1,6 +1,8 @@
 #pragma once
 
 #include <vector>
+#include <memory>
+
 #include "body.h"
 
 template <typename T>
diff --git a/src/octree.cpp b/src/octree.cpp
index 6a9af6f..1292458 100644
--- a/src/octree.cpp
+++ b/src/octree.cpp
@@ -1,6 +1,6 @@
 #include "./../include/octree.h"
 #include <array>
-#include <iostream>
+#include <memory>
 
 template <typename T>
 Octree<T>::Octree(std::array<float, 2>& xCoords, std::array<float, 2>& yCoords, std::array<float, 2>& zCoords)