From 1a24139359e6b65dcf66696ae49fb1294babe160 Mon Sep 17 00:00:00 2001
From: Nikita Sivukhin <sivukhin@turso.tech>
Date: Tue, 14 Oct 2025 13:23:22 +0400
Subject: [PATCH] fix limit for order by queries with heap-sort style execution

---
 core/translate/order_by.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/core/translate/order_by.rs b/core/translate/order_by.rs
index 09de9c3d3..7a91bc160 100644
--- a/core/translate/order_by.rs
+++ b/core/translate/order_by.rs
@@ -239,7 +239,11 @@ pub fn emit_order_by(
         plan,
         start_reg,
         t_ctx.limit_ctx,
-        Some(sort_loop_end_label),
+        if !use_heap_sort {
+            Some(sort_loop_end_label)
+        } else {
+            None
+        },
     )?;
 
     program.resolve_label(sort_loop_next_label, program.offset());