diff --git a/include/llvm/Analysis/DSGraph.h b/include/llvm/Analysis/DSGraph.h
index 6fa4d6eecac..cccf31f8f3b 100644
--- a/include/llvm/Analysis/DSGraph.h
+++ b/include/llvm/Analysis/DSGraph.h
@@ -436,6 +436,8 @@ public:
         DSNodeHandle &getCallee()           { return Callee; }
   const DSNodeHandle &getRetVal()     const { return RetVal; }
   const DSNodeHandle &getCallee()     const { return Callee; }
+  void setCallee(const DSNodeHandle &H) { Callee = H; }
+
   unsigned            getNumPtrArgs() const { return CallArgs.size(); }
 
   Function           *getResolvingCaller() const { return ResolvingCaller; }
diff --git a/include/llvm/Analysis/DataStructure/DSGraph.h b/include/llvm/Analysis/DataStructure/DSGraph.h
index 6fa4d6eecac..cccf31f8f3b 100644
--- a/include/llvm/Analysis/DataStructure/DSGraph.h
+++ b/include/llvm/Analysis/DataStructure/DSGraph.h
@@ -436,6 +436,8 @@ public:
         DSNodeHandle &getCallee()           { return Callee; }
   const DSNodeHandle &getRetVal()     const { return RetVal; }
   const DSNodeHandle &getCallee()     const { return Callee; }
+  void setCallee(const DSNodeHandle &H) { Callee = H; }
+
   unsigned            getNumPtrArgs() const { return CallArgs.size(); }
 
   Function           *getResolvingCaller() const { return ResolvingCaller; }