The new code calls self.graphql_span.__enter__() at line 189 and self.graphql_span.__exit__(None, None, None) at line 204, but if an exception is thrown into the generator at the yield statement (line 196), the code after yield never executes. This leaves the span unfinished and corrupts the scope because __exit__ restores the old span. Additionally, passing (None, None, None) to __exit__ prevents the span from being marked with INTERNAL_ERROR status when an exception occurs.