diff --git a/tools/src/main/scala/com/nvidia/spark/rapids/tool/qualification/QualOutputWriter.scala b/tools/src/main/scala/com/nvidia/spark/rapids/tool/qualification/QualOutputWriter.scala
index a30ffa201e8..53303c6c5e2 100644
--- a/tools/src/main/scala/com/nvidia/spark/rapids/tool/qualification/QualOutputWriter.scala
+++ b/tools/src/main/scala/com/nvidia/spark/rapids/tool/qualification/QualOutputWriter.scala
@@ -123,7 +123,7 @@ class QualOutputWriter(outputDir: String, reportReadSchema: Boolean,
       sortedAsc.reverse
     }
     val finalSums = sorted.take(numOutputRows)
-    sorted.foreach { estInfo =>
+    finalSums.foreach { estInfo =>
       val wStr = QualOutputWriter.constructPerSqlSummaryInfo(estInfo, headersAndSizes,
         appIdMaxSize, TEXT_DELIMITER, true, maxSQLDescLength)
       writer.write(wStr)
diff --git a/tools/src/test/scala/com/nvidia/spark/rapids/tool/qualification/QualificationSuite.scala b/tools/src/test/scala/com/nvidia/spark/rapids/tool/qualification/QualificationSuite.scala
index 100ebd4b840..98688175c0d 100644
--- a/tools/src/test/scala/com/nvidia/spark/rapids/tool/qualification/QualificationSuite.scala
+++ b/tools/src/test/scala/com/nvidia/spark/rapids/tool/qualification/QualificationSuite.scala
@@ -312,7 +312,7 @@ class QualificationSuite extends FunSuite with BeforeAndAfterEach with Logging {
       try {
         val lines = persqlInputSource.getLines
         // 4 lines of header and footer, limit is 2
-        assert(lines.size == (4 + 17))
+        assert(lines.size == (4 + 2))
       } finally {
         persqlInputSource.close()
       }