diff --git a/presto-main/src/main/java/io/prestosql/snapshot/MarkerSplitSource.java b/presto-main/src/main/java/io/prestosql/snapshot/MarkerSplitSource.java
index 9780d0a89..3053146dd 100644
--- a/presto-main/src/main/java/io/prestosql/snapshot/MarkerSplitSource.java
+++ b/presto-main/src/main/java/io/prestosql/snapshot/MarkerSplitSource.java
@@ -374,6 +374,8 @@ public class MarkerSplitSource
     public void close()
     {
         announcer.deactivateSplitSource(this);
+        // In case of closing split source before source is exhausted. (e.g. query with limit function).
+        sourceExhausted = true;
         source.close();
     }