diff --git a/firestore/query.go b/firestore/query.go index 0c0d8ed3b75c..6d92d17a1862 100644 --- a/firestore/query.go +++ b/firestore/query.go @@ -1054,9 +1054,9 @@ func (it *queryDocumentIterator) next() (_ *DocumentSnapshot, err error) { client := it.q.c if it.streamClient == nil { it.ctx = trace.StartSpan(it.ctx, "cloud.google.com/go/firestore.Query.RunQuery") - defer func() { + defer func() { if errors.Is(err, iterator.Done) { - trace.EndSpan(it.ctx, nil) + trace.EndSpan(it.ctx, nil) } else { trace.EndSpan(it.ctx, err) }