diff --git a/logservice/logpuller/subscription_client.go b/logservice/logpuller/subscription_client.go
index 0c04781d..ad5dee95 100644
--- a/logservice/logpuller/subscription_client.go
+++ b/logservice/logpuller/subscription_client.go
@@ -324,7 +324,6 @@ func (s *SubscriptionClient) Unsubscribe(subID SubscriptionID) {
 		log.Warn("unknown subscription", zap.Uint64("subscriptionID", uint64(subID)))
 		return
 	}
-	s.ds.RemovePath(rt.subID)
 	s.setTableStopped(rt)
 
 	log.Info("unsubscribe span success",
@@ -398,6 +397,7 @@ func (s *SubscriptionClient) onTableDrained(rt *subscribedSpan) {
 
 	s.totalSpans.Lock()
 	defer s.totalSpans.Unlock()
+	s.ds.RemovePath(rt.subID)
 	delete(s.totalSpans.spanMap, rt.subID)
 }