1 files changed, 17 insertions, 1 deletions

diff --git a/trace-graph.c b/trace-graph.c
index dd99b4b..b2ff3b2 100644
--- a/trace-graph.c
+++ b/trace-graph.c
@@ -661,7 +661,7 @@ static void draw_cpu(struct graph_info *ginfo, gint cpu,
         struct record *record;
         static GdkGC *gc;
         guint64 ts;
-        gint last_pid = 0;
+        gint last_pid = -1;
         gint last_x = 0;
         gint pid;
         gint x;
@@ -690,6 +690,10 @@ static void draw_cpu(struct graph_info *ginfo, gint cpu,
                         pid = pevent_data_pid(ginfo->pevent, record);
 
                 if (last_pid != pid) {
+
+                        if (last_pid < 0)
+                                last_pid = pid;
+
                         if (last_pid) {
                                 gdk_draw_line(ginfo->curr_pixmap, gc,
                                               last_x, CPU_TOP(cpu),
@@ -710,6 +714,18 @@ static void draw_cpu(struct graph_info *ginfo, gint cpu,
                 free(record);
         }
 
+
+        if (last_pid > 0) {
+                x = ginfo->draw_width;
+
+                gdk_draw_line(ginfo->curr_pixmap, gc,
+                              last_x, CPU_TOP(cpu),
+                              x, CPU_TOP(cpu));
+                gdk_draw_line(ginfo->curr_pixmap, gc,
+                              last_x, CPU_BOTTOM(cpu),
+                              x, CPU_BOTTOM(cpu));
+        }
+
         if (record)
                 free(record);
 }