diff --git a/llvm/lib/Analysis/DataStructure/Printer.cpp b/llvm/lib/Analysis/DataStructure/Printer.cpp
index a5cb4da..e25bc4c 100644
--- a/llvm/lib/Analysis/DataStructure/Printer.cpp
+++ b/llvm/lib/Analysis/DataStructure/Printer.cpp
@@ -249,7 +249,7 @@
           << Gr.getGraphSize() << "+" << NumCalls << "]\n";
       }
 
-      unsigned GraphSize = std::distance(Gr.node_begin(), Gr.node_end());
+      unsigned GraphSize = Gr.getGraphSize();
       if (MaxGraphSize < GraphSize) MaxGraphSize = GraphSize;
 
       for (DSGraph::node_iterator NI = Gr.node_begin(), E = Gr.node_end();