diff --git a/lib/Frontend/AnalysisConsumer.cpp b/lib/Frontend/AnalysisConsumer.cpp
index 468bbf6..7d1dc05 100644
--- a/lib/Frontend/AnalysisConsumer.cpp
+++ b/lib/Frontend/AnalysisConsumer.cpp
@@ -158,7 +158,7 @@
     }
     
     virtual CFG* getCFG() {
-      if (!cfg) cfg.reset(CFG::buildCFG(getBody()));
+      if (!cfg) cfg.reset(CFG::buildCFG(getBody(), &getContext()));
       return cfg.get();
     }