diff --git a/flat_mutation_reader.hh b/flat_mutation_reader.hh
index 9d15f63e42..3c4b610816 100644
--- a/flat_mutation_reader.hh
+++ b/flat_mutation_reader.hh
@@ -585,7 +585,7 @@ future<> consume_mutation_fragments_until(
     return do_until([stop] { return stop(); }, [&r, stop, consume_mf, consume_eos, timeout] {
         while (!r.is_buffer_empty()) {
             consume_mf(r.pop_mutation_fragment());
-            if (stop()) {
+            if (stop() || need_preempt()) {
                 return make_ready_future<>();
             }
         }