diff --git a/llvm/lib/Analysis/DataStructure/Local.cpp b/llvm/lib/Analysis/DataStructure/Local.cpp
index 93c534a..2df0fd4 100644
--- a/llvm/lib/Analysis/DataStructure/Local.cpp
+++ b/llvm/lib/Analysis/DataStructure/Local.cpp
@@ -394,7 +394,8 @@
 
 void GraphBuilder::visitFreeInst(FreeInst &FI) {
   // Mark that the node is written to...
-  getValueDest(*FI.getOperand(0)).getNode()->NodeType |= DSNode::Modified;
+  getValueDest(*FI.getOperand(0)).getNode()->NodeType
+    |= DSNode::Modified | DSNode::HeapNode;
 }
 
 /// Handle casts...