diff --git a/paimon-core/src/main/java/org/apache/paimon/utils/TagManager.java b/paimon-core/src/main/java/org/apache/paimon/utils/TagManager.java
index 2aaefa93da776..f609a281a388e 100644
--- a/paimon-core/src/main/java/org/apache/paimon/utils/TagManager.java
+++ b/paimon-core/src/main/java/org/apache/paimon/utils/TagManager.java
@@ -372,7 +372,7 @@ public List<String> allTagNames() {
         return tags().values().stream().flatMap(Collection::stream).collect(Collectors.toList());
     }
 
-    /** askwang-todo: 二分查找加速 taggedSnapshot 查询速度. */
+    /** askwang-done: 二分查找加速 taggedSnapshot 查询速度. A: 内存查找很快，无需优化. */
     private int findIndex(Snapshot taggedSnapshot, List<Snapshot> taggedSnapshots) {
         for (int i = 0; i < taggedSnapshots.size(); i++) {
             if (taggedSnapshot.id() == taggedSnapshots.get(i).id()) {
@@ -452,7 +452,7 @@ public static int findPreviousTag(List<Snapshot> taggedSnapshots, long targetSna
     /** 减少不重叠的 snapshot 范围 [begin, end): [10, 15) taggedSnapshots: [7, 9, 11, 12]. */
     private static int findNextOrEqualTagAskwang(
             List<Snapshot> taggedSnapshots, long targetSnapshotId) {
-        for (int i = 0; i < targetSnapshotId; i++) {
+        for (int i = 0; i < taggedSnapshots.size(); i++) {
             if (taggedSnapshots.get(i).id() >= targetSnapshotId) {
                 return i;
             }
diff --git a/paimon-spark/paimon-spark-common/src/test/scala/org/apache/paimon/spark/procedure/ExpireSnapshotsProcedureTest.scala b/paimon-spark/paimon-spark-common/src/test/scala/org/apache/paimon/spark/procedure/ExpireSnapshotsProcedureTest.scala
index d92fdb2d90126..4c615d766b8ce 100644
--- a/paimon-spark/paimon-spark-common/src/test/scala/org/apache/paimon/spark/procedure/ExpireSnapshotsProcedureTest.scala
+++ b/paimon-spark/paimon-spark-common/src/test/scala/org/apache/paimon/spark/procedure/ExpireSnapshotsProcedureTest.scala
@@ -139,7 +139,7 @@ class ExpireSnapshotsProcedureTest extends PaimonSparkTestBase with StreamTest {
 
   // min maybe large than maxExclusive
   test("Paimon Procedure: max delete") {
-    failAfter(streamingTimeout) {
+    // failAfter(streamingTimeout) {
       withTempDir {
         checkpointDir =>
           // define a change-log table and test `forEachBatch` api
@@ -207,15 +207,12 @@ class ExpireSnapshotsProcedureTest extends PaimonSparkTestBase with StreamTest {
             inputData.addData((10, "b"))
             stream.processAllAvailable()
 
-            // expire assert throw exception
-            assertThrows[IllegalArgumentException] {
-              spark.sql(
+            spark.sql(
                 "CALL paimon.sys.expire_snapshots(table => 'test.T', retain_max => 5, retain_min => 2, max_deletes => 4)")
-            }
           } finally {
             stream.stop()
           }
       }
-    }
+    // }
   }
 }