diff --git a/lib/Analysis/DataStructure/DataStructureStats.cpp b/lib/Analysis/DataStructure/DataStructureStats.cpp
index ffd560b2457..f033b99b895 100644
--- a/lib/Analysis/DataStructure/DataStructureStats.cpp
+++ b/lib/Analysis/DataStructure/DataStructureStats.cpp
@@ -104,7 +104,7 @@ DSNode *DSGraphStats::getNodeForValue(Value *V) {
 
 bool DSGraphStats::isNodeForValueCollapsed(Value *V) {
   if (DSNode *N = getNodeForValue(V))
-    return N->isNodeCompletelyFolded();
+    return N->isNodeCompletelyFolded() || N->isIncomplete();
   return false;
 }