diff --git a/cachegrind/cg_main.c b/cachegrind/cg_main.c
index 2b36d43..02fccc1 100644
--- a/cachegrind/cg_main.c
+++ b/cachegrind/cg_main.c
@@ -30,7 +30,7 @@
 */
 
 #include "tool.h"
-//#include "vg_profile.c"
+#include "pub_tool_mallocfree.h"
 
 #include "cg_arch.h"
 #include "cg_sim.c"