diff --git a/math_coding/flows/math_standard_flow/flow.dag.yaml b/math_coding/flows/math_standard_flow/flow.dag.yaml
index 6ca658b98..8cb21cd17 100644
--- a/math_coding/flows/math_standard_flow/flow.dag.yaml
+++ b/math_coding/flows/math_standard_flow/flow.dag.yaml
@@ -39,8 +39,10 @@ nodes:
     type: code
     path: ask_llm.jinja2
   inputs:
-    deployment_name: gpt-4o
-    model: gpt-4o
+    # This is to easily switch between openai and azure openai.
+    # deployment_name is required by azure openai, model is required by openai.
+    deployment_name: gpt-35-turbo
+    model: gpt-3.5-turbo
     question: ${inputs.math_question}
     examples: ${math_example.output}
   provider: AzureOpenAI