1 files changed, 4 insertions, 3 deletions
diff --git a/ui/src/tracks/heap_profile/frontend.ts b/ui/src/tracks/heap_profile/frontend.ts
index 8eaba5e7c..86d177e84 100644
--- a/ui/src/tracks/heap_profile/frontend.ts
+++ b/ui/src/tracks/heap_profile/frontend.ts
@@ -118,10 +118,11 @@ class HeapProfileTrack extends Track<Config, Data> {
 
     if (index !== -1) {
       const ts = data.tsStarts[index];
+      const type = data.types[index];
       globals.dispatch(Actions.showHeapProfileFlamegraph(
-          {id: index, upid: this.config.upid, ts}));
-      globals.makeSelection(
-          Actions.selectHeapProfile({id: index, upid: this.config.upid, ts}));
+          {id: index, upid: this.config.upid, ts, type}));
+      globals.makeSelection(Actions.selectHeapProfile(
+          {id: index, upid: this.config.upid, ts, type}));
       return true;
     }
     return false;