Release 18.05
diff --git a/tests/InferenceTest.inl b/tests/InferenceTest.inl
index 83a9945..a36e231 100644
--- a/tests/InferenceTest.inl
+++ b/tests/InferenceTest.inl
@@ -307,6 +307,7 @@
                     modelParams.m_InputTensorShape = inputTensorShape;
                     modelParams.m_IsModelBinary = isModelBinary;
                     modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice;
+                    modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel;
 
                     return std::make_unique<InferenceModel>(modelParams);
             });