1 files changed, 4 insertions, 2 deletions

diff --git a/trace-plot-task.c b/trace-plot-task.c
index b36d565..a5b8d79 100644
--- a/trace-plot-task.c
+++ b/trace-plot-task.c
@@ -632,7 +632,8 @@ void graph_plot_init_tasks(struct graph_info *ginfo)
         task_info->pid = pid;
 
         snprintf(label, 100, "TASK %d", pid);
-        trace_graph_plot_insert(ginfo, 1, label, &task_plot_cb, task_info);
+        trace_graph_plot_insert(ginfo, 1, label, PLOT_TYPE_TASK,
+                                &task_plot_cb, task_info);
 }
 
 void graph_plot_task(struct graph_info *ginfo, int pid, int pos)
@@ -650,7 +651,8 @@ void graph_plot_task(struct graph_info *ginfo, int pid, int pos)
         len = strlen(comm) + 100;
         label = malloc_or_die(len);
         snprintf(label, len, "%s-%d", comm, pid);
-        plot = trace_graph_plot_insert(ginfo, pos, label, &task_plot_cb, task_info);
+        plot = trace_graph_plot_insert(ginfo, pos, label, PLOT_TYPE_TASK,
+                                       &task_plot_cb, task_info);
         free(label);
 
         trace_graph_plot_add_all_recs(ginfo, plot);