diff --git a/lib/CodeGen/CodeGenFunction.cpp b/lib/CodeGen/CodeGenFunction.cpp
index 679a0f7..e153ad9 100644
--- a/lib/CodeGen/CodeGenFunction.cpp
+++ b/lib/CodeGen/CodeGenFunction.cpp
@@ -102,10 +102,11 @@
   assert(!verifyFunction(*CurFn) && "Generated function is not well formed.");
 }
 
-void CodeGenFunction::GenerateCode(const FunctionDecl *FD) {
+void CodeGenFunction::GenerateCode(const FunctionDecl *FD,
+                                   llvm::Function *Fn) {
   CurFuncDecl = FD;
   FnRetTy = FD->getResultType();
-  CurFn = cast<llvm::Function>(CGM.GetAddrOfFunctionDecl(FD, true));
+  CurFn = Fn;
   assert(CurFn->isDeclaration() && "Function already has body?");
 
   llvm::BasicBlock *EntryBB = llvm::BasicBlock::Create("entry", CurFn);