1 files changed, 5 insertions, 7 deletions
diff --git a/lib/VMCore/Dominators.cpp b/lib/VMCore/Dominators.cpp
index 0f68bb4d02..76ad59b354 100644
--- a/lib/VMCore/Dominators.cpp
+++ b/lib/VMCore/Dominators.cpp
@@ -211,15 +211,13 @@ bool ImmediateDominators::runOnFunction(Function &F) {
 }
 
 void ImmediateDominatorsBase::print(std::ostream &o) const {
-  for (const_iterator I = begin(), E = end(); I != E; ++I) {
+  Function *F = getRoots()[0]->getParent();
+  for (Function::iterator I = F->begin(), E = F->end(); I != E; ++I) {
     o << "  Immediate Dominator For Basic Block:";
-    if (I->first)
-      WriteAsOperand(o, I->first, false);
-    else
-      o << " <<exit node>>";
+    WriteAsOperand(o, I, false);
     o << " is:";
-    if (I->second)
-      WriteAsOperand(o, I->second, false);
+    if (BasicBlock *ID = get(I))
+      WriteAsOperand(o, ID, false);
     else
       o << " <<exit node>>";
     o << "\n";