diff --git a/vcr/cassette.py b/vcr/cassette.py index fad0d25d..5a189e32 100644 --- a/vcr/cassette.py +++ b/vcr/cassette.py @@ -125,7 +125,7 @@ def _handle_generator(self, fn): duration of the generator. """ with self as cassette: - yield from fn(cassette) + return (yield from fn(cassette)) def _handle_function(self, fn): with self as cassette: