postgres
diff --git a/‎src/backend/executor/nodeIncrementalSort.c
Copy file name to clipboardExpand all lines: src/backend/executor/nodeIncrementalSort.c
+7Lines changed: 7 additions & 0 deletions b/‎src/backend/executor/nodeIncrementalSort.c
Copy file name to clipboardExpand all lines: src/backend/executor/nodeIncrementalSort.c
+7Lines changed: 7 additions & 0 deletions
diff --git a/‎src/test/regress/expected/incremental_sort.out
Copy file name to clipboardExpand all lines: src/test/regress/expected/incremental_sort.out
+11Lines changed: 11 additions & 0 deletions b/‎src/test/regress/expected/incremental_sort.out
Copy file name to clipboardExpand all lines: src/test/regress/expected/incremental_sort.out
+11Lines changed: 11 additions & 0 deletions
diff --git a/‎src/test/regress/sql/incremental_sort.sql
Copy file name to clipboardExpand all lines: src/test/regress/sql/incremental_sort.sql
+3Lines changed: 3 additions & 0 deletions b/‎src/test/regress/sql/incremental_sort.sql
Copy file name to clipboardExpand all lines: src/test/regress/sql/incremental_sort.sql
+3Lines changed: 3 additions & 0 deletions
@@ -394,6 +394,13 @@ switchToPresortedPrefixMode(PlanState *pstate)
 				 * current prefix key group.
 				 */
 				ExecClearTuple(node->group_pivot);
+
+				/*
+				 * Also make sure we take the didn't-consume-all-the-tuples
+				 * path below, even if this happened to be the last tuple of
+				 * the batch.
+				 */
+				lastTuple = false;
 				break;
 			}
 		}
 
@@ -675,6 +675,17 @@ select * from (select * from t order by a) s order by a, b limit 70;
  9 | 70
 (70 rows)
 
+-- Checks case where we hit a group boundary at the last tuple of a batch.
+select * from (select * from t order by a) s order by a, b limit 5;
+ a | b 
+---+---
+ 1 | 1
+ 2 | 2
+ 3 | 3
+ 4 | 4
+ 9 | 5
+(5 rows)
+
 -- Test rescan.
 begin;
 -- We force the planner to choose a plan with incremental sort on the right side
 
@@ -149,6 +149,9 @@ insert into t(a, b) select (case when i < 5 then i else 9 end), i from generate_
 analyze t;
 explain (costs off) select * from (select * from t order by a) s order by a, b limit 70;
 select * from (select * from t order by a) s order by a, b limit 70;
+-- Checks case where we hit a group boundary at the last tuple of a batch.
+select * from (select * from t order by a) s order by a, b limit 5;
+
 -- Test rescan.
 begin;
 -- We force the planner to choose a plan with incremental sort on the right side