diff --git a/lib/AST/CFG.cpp b/lib/AST/CFG.cpp
index 4b7085a..35eaa14 100644
--- a/lib/AST/CFG.cpp
+++ b/lib/AST/CFG.cpp
@@ -840,7 +840,7 @@
   
   // Now create the true branch.
   Succ = ConditionBlock;
-  CFGBlock* BodyBlock = addStmt(S->getBody());
+  CFGBlock* BodyBlock = Visit(S->getBody());
   FinishBlock(BodyBlock);
   
   // Connect up the condition block