diff --git a/src/language/ASTBuilder.cpp b/src/language/ASTBuilder.cpp
index 8273412a39dbca5f5592df878c30ee9c6d17af6a..709953c6766a6eab38aad1f3220ee058e091dda9 100644
--- a/src/language/ASTBuilder.cpp
+++ b/src/language/ASTBuilder.cpp
@@ -5,6 +5,7 @@ using namespace TAO_PEGTL_NAMESPACE;
 #include <language/ASTNode.hpp>
 #include <language/PEGGrammar.hpp>
 #include <language/SymbolTable.hpp>
+#include <utils/PugsAssert.hpp>
 
 #include <pegtl/contrib/parse_tree.hpp>
 
diff --git a/src/language/ASTBuilder.hpp b/src/language/ASTBuilder.hpp
index 9658eaaac6b440c58aed730df10fc67c95ccb351..2691a724d70ce40c99338441269280e0ac4c40a9 100644
--- a/src/language/ASTBuilder.hpp
+++ b/src/language/ASTBuilder.hpp
@@ -3,6 +3,8 @@
 
 #include <language/ASTNode.hpp>
 
+#include <memory>
+
 struct ASTBuilder
 {
  private: