1 files changed, 4 insertions, 5 deletions
diff --git a/node/BaseNode.hh b/node/BaseNode.hh
index 60e25bd..4a3746c 100644
--- a/node/BaseNode.hh
+++ b/node/BaseNode.hh
@@ -3,17 +3,16 @@
 
 #include <llvm/IR/Value.h>
 
-namespace pcc {
-using Value = llvm::Value*;
-class Driver;
+#include "Context.hh"
 
+namespace pcc {
 class BaseNode {
   public:
     virtual Value CodeGen() = 0;
-    BaseNode(Driver* driver);
+    BaseNode(Context* context);
 
   protected:
-    Driver* driver_;
+    Context* context_;
 };
 } // namespace pcc