diff --git a/tests/benchmarks/test_execute_with_extensions.py b/tests/benchmarks/test_execute_with_extensions.py index 55a0af4a82..fd93f141f8 100644 --- a/tests/benchmarks/test_execute_with_extensions.py +++ b/tests/benchmarks/test_execute_with_extensions.py @@ -20,7 +20,7 @@ def get_results(self) -> AwaitableOrValue[Dict[str, Any]]: class ResolveExtension(SchemaExtension): async def resolve(self, _next, root, info, *args: Any, **kwargs: Any) -> Any: result = _next(root, info, *args, **kwargs) - if self.should_await(_next): + if _next._is_async: # type: ignore result = await result return result