From e896bbb78cef9cc084667cf554418b6a82cb9ae7 Mon Sep 17 00:00:00 2001
From: Li Haoyi <haoyi.sg@gmail.com>
Date: Sat, 18 Mar 2023 07:06:54 +0800
Subject: [PATCH] .

---
 bench/src-jvm/Main.scala | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/bench/src-jvm/Main.scala b/bench/src-jvm/Main.scala
index 2411e88e4..30d2d35a4 100644
--- a/bench/src-jvm/Main.scala
+++ b/bench/src-jvm/Main.scala
@@ -95,8 +95,9 @@ object Main{
         val before = new String(java.nio.file.Files.readAllBytes(p))
         def split(s: String) = s.linesIterator.map(_.drop(2).dropRight(2).split(" *\\| *")).toList
         println(
-          split(prettyResults).zip(split(before)).map {
-            case (afterLine, beforeLine) =>
+          split(prettyResults).zip(split(before)).zipWithIndex.map {
+            case ((afterLine, beforeLine), 0 | 1) => afterLine
+            case ((afterLine, beforeLine), _) =>
               afterLine.zip(beforeLine) match {
                 case Array(head, tail@_*) =>
                   Array(head._1) ++ tail.map { case (a, b) =>