diff --git a/sweepbatcher/sweep_batcher_test.go b/sweepbatcher/sweep_batcher_test.go
index 590d83ae..955b72ad 100644
--- a/sweepbatcher/sweep_batcher_test.go
+++ b/sweepbatcher/sweep_batcher_test.go
@@ -2,6 +2,7 @@ package sweepbatcher
 
 import (
 	"context"
+	"database/sql"
 	"errors"
 	"fmt"
 	"os"
@@ -3997,6 +3998,9 @@ func testSweepBatcherCloseDuringAdding(t *testing.T, store testStore,
 			if errors.Is(err, context.Canceled) {
 				break
 			}
+			if errors.Is(err, sql.ErrTxDone) {
+				break
+			}
 			require.NoError(t, err)
 		}
 	}()