| Index: recipe_engine/simulation_test.py
|
| diff --git a/recipe_engine/simulation_test.py b/recipe_engine/simulation_test.py
|
| index 17c1d1e511689cdb38340f1af3a4585b640e46df..e82faabfefc2650a5b5be6de1a1bbb8314089bbf 100644
|
| --- a/recipe_engine/simulation_test.py
|
| +++ b/recipe_engine/simulation_test.py
|
| @@ -81,10 +81,7 @@ def RunRecipe(test_data):
|
| config_types.ResetTostringFns()
|
|
|
| annotator = SimulationAnnotatorStreamEngine()
|
| - stream_engine = stream.ProductStreamEngine(
|
| - stream.StreamEngineInvariants(),
|
| - annotator)
|
| - with stream_engine:
|
| + with stream.StreamEngineInvariants.wrap(annotator) as stream_engine:
|
| step_runner = step_runner.SimulationStepRunner(stream_engine, test_data,
|
| annotator)
|
|
|
|
|