diff --git a/lib/Transforms/IPO/InlineSimple.cpp b/lib/Transforms/IPO/InlineSimple.cpp
index 53add2ddfb8..f1e92e40120 100644
--- a/lib/Transforms/IPO/InlineSimple.cpp
+++ b/lib/Transforms/IPO/InlineSimple.cpp
@@ -18,7 +18,8 @@
 //
 //===----------------------------------------------------------------------===//
 
-#include "llvm/Transforms/FunctionInlining.h"
+#include "llvm/Transforms/IPO.h"
+#include "llvm/Transforms/Utils/Cloning.h"
 #include "llvm/Module.h"
 #include "llvm/Pass.h"
 #include "llvm/iTerminators.h"
diff --git a/lib/Transforms/Utils/CloneFunction.cpp b/lib/Transforms/Utils/CloneFunction.cpp
index 866b8c3bbe4..0a2c41f7856 100644
--- a/lib/Transforms/Utils/CloneFunction.cpp
+++ b/lib/Transforms/Utils/CloneFunction.cpp
@@ -1,10 +1,10 @@
 
+
+
 // FIXME: document
 
-#include "llvm/Transforms/Utils/CloneFunction.h"
+#include "llvm/Transforms/Utils/Cloning.h"
 #include "llvm/Function.h"
-#include "llvm/BasicBlock.h"
-#include "llvm/Instruction.h"
 #include <map>
 
 // FIXME: This should be merged with FunctionInlining