diff --git a/poolalloc/include/poolalloc/PoolAllocate.h b/poolalloc/include/poolalloc/PoolAllocate.h
index 449094d..f2d38fe 100644
--- a/poolalloc/include/poolalloc/PoolAllocate.h
+++ b/poolalloc/include/poolalloc/PoolAllocate.h
@@ -218,6 +218,14 @@
 #endif
   }
 
+  virtual DSGraph & getDSGraph (const Function & F) const {
+    return ECGraphs->getDSGraph (F);
+  }
+
+  virtual DSGraph & getGlobalsGraph () const {
+    return ECGraphs->getGlobalsGraph ();
+  }
+
   virtual Value * getGlobalPool (const DSNode * Node) {
     std::map<const DSNode *, Value *>::iterator I = GlobalNodes.find (Node);
     if (I == GlobalNodes.end())