diff --git a/tvlm/il.h b/tvlm/il.h
index dadaeb5..7ae38f9 100644
--- a/tvlm/il.h
+++ b/tvlm/il.h
@@ -5,6 +5,8 @@
 #include "common/helpers.h"
 #include "common/ast.h"
 
+#include <memory>
+
 namespace tvlm {
 
     using ASTBase = tiny::ASTBase;
@@ -369,7 +371,7 @@ class BasicBlock {
         }
 
     private:
-        friend class Builder;
+        friend class ILBuilder;
 
         ASTBase const * ast_;