diff --git a/lib/Sema/ParseAST.cpp b/lib/Sema/ParseAST.cpp
index cff3ef7..9199179 100644
--- a/lib/Sema/ParseAST.cpp
+++ b/lib/Sema/ParseAST.cpp
@@ -46,7 +46,7 @@
   // Initialize the parser.
   P.Initialize();
   
-  Consumer->Initialize(Context);
+  Consumer->InitializeTU(TU);
   
   Parser::DeclTy *ADecl;