diff --git a/include/llvm/Analysis/DSGraph.h b/include/llvm/Analysis/DSGraph.h
index 8e62a361128..285724a5769 100644
--- a/include/llvm/Analysis/DSGraph.h
+++ b/include/llvm/Analysis/DSGraph.h
@@ -157,7 +157,7 @@ public:
 
   // Iterator for graph interface...
   typedef DSNodeIterator iterator;
-  inline iterator begin();   // Defined in DataStructureGraph.h
+  inline iterator begin();   // Defined in DSGraphTraits.h
   inline iterator end();
 
   //===--------------------------------------------------
diff --git a/include/llvm/Analysis/DataStructure/DSGraph.h b/include/llvm/Analysis/DataStructure/DSGraph.h
index 8e62a361128..285724a5769 100644
--- a/include/llvm/Analysis/DataStructure/DSGraph.h
+++ b/include/llvm/Analysis/DataStructure/DSGraph.h
@@ -157,7 +157,7 @@ public:
 
   // Iterator for graph interface...
   typedef DSNodeIterator iterator;
-  inline iterator begin();   // Defined in DataStructureGraph.h
+  inline iterator begin();   // Defined in DSGraphTraits.h
   inline iterator end();
 
   //===--------------------------------------------------