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) |