diff --git a/vmstat.c b/vmstat.c
index 53677628..a5c62d6d 100644
--- a/vmstat.c
+++ b/vmstat.c
@@ -464,8 +464,8 @@ static int diskpartition_format(const char *partition_name)
         if (infinite_updates || i+1 < num_updates)
             sleep(sleep_time);
     }
-    return 0;
     procps_diskstat_unref(&disk_stat);
+    return 0;
 }
 
 static void diskheader(void)