diff --git a/docs/notebooks/nvidia/tool_calling/config.py b/docs/notebooks/nvidia/tool_calling/config.py
index e7ebf924c..009c63e3c 100644
--- a/docs/notebooks/nvidia/tool_calling/config.py
+++ b/docs/notebooks/nvidia/tool_calling/config.py
@@ -9,6 +9,9 @@ NDS_URL = "http://data-store.test:3000" # Data Store
 NEMO_URL = "http://nemo.test:3000" # Customizer, Evaluator, Guardrails
 NIM_URL = "http://nim.test:3000" # NIM
 
+# (Required) Configure the base model. Must be one supported by the NeMo Customizer deployment!
+BASE_MODEL = "meta-llama/Llama-3.2-1B-Instruct"
+
 # (Required) Hugging Face Token
 HF_TOKEN = ""
 
@@ -19,9 +22,6 @@ NDS_TOKEN = "token"
 NMS_NAMESPACE = "nvidia-tool-calling-tutorial"
 DATASET_NAME = "xlam-ft-dataset-1"
 
-# (Optional) Configure the base model. Must be one supported by the NeMo Customizer deployment!
-BASE_MODEL = "meta-llama/Llama-3.2-1B-Instruct"
-
 # (Optional) NVIDIA User ID - currently unused
 USER_ID = ""
 # (Optional) Entity Store Project ID. Modify if you've created a project in Entity Store that you'd