1 files changed, 3 insertions, 2 deletions
diff --git a/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp b/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp
index 3328339318..06196475bf 100644
--- a/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp
+++ b/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp
@@ -40,7 +40,8 @@ int main(int argc, char* argv[])
             {
                 return make_unique<MobileNetSsdTestCaseProvider<Model>>(
                     [&]
-                    (typename Model::CommandLineOptions modelOptions)
+                    (const InferenceTestOptions& commonOptions,
+                     typename Model::CommandLineOptions modelOptions)
                     {
                         if (!ValidateDirectory(modelOptions.m_ModelDir))
                         {
@@ -63,7 +64,7 @@ int main(int argc, char* argv[])
                         modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel;
                         modelParams.m_EnableFp16TurboMode            = modelOptions.m_EnableFp16TurboMode;
 
-                        return std::make_unique<Model>(modelParams);
+                        return std::make_unique<Model>(modelParams, commonOptions.m_EnableProfiling);
                 });
             });
     }