diff --git a/tests/core/pyspec/eth2spec/test/context.py b/tests/core/pyspec/eth2spec/test/context.py index a41af7dd2..d20446987 100644 --- a/tests/core/pyspec/eth2spec/test/context.py +++ b/tests/core/pyspec/eth2spec/test/context.py @@ -602,13 +602,13 @@ def with_config_overrides(config_overrides, emitted_fork=None, emit=True): output_config = output kw['phases'] = phases - # Run the function - out = fn(*args, spec=spec, **kw) - # Emit requested spec (with overrides) if emit: yield 'config', 'cfg', output_config + # Run the function + out = fn(*args, spec=spec, **kw) + # If it's not returning None like a normal test function, # it's generating things, and we need to complete it before setting back the config. if out is not None: