diff --git a/internal/querynodev2/delegator/distribution_test.go b/internal/querynodev2/delegator/distribution_test.go
index cae37f154b..8dae155810 100644
--- a/internal/querynodev2/delegator/distribution_test.go
+++ b/internal/querynodev2/delegator/distribution_test.go
@@ -145,10 +145,12 @@ func (s *DistributionSuite) TestAddDistribution() {
 			s.SetupTest()
 			defer s.TearDownTest()
 			s.dist.AddGrowing(tc.growing...)
+			_, _, version := s.dist.GetCurrent()
 			_, signal := s.dist.AddDistributions(tc.input...)
 			sealed, _ := s.dist.Peek()
 			s.compareSnapshotItems(tc.expected, sealed)
 			s.Equal(tc.expectedSignalClosed, s.isClosedCh(signal))
+			s.dist.FinishUsage(version)
 		})
 	}
 }