diff --git a/orttraining/orttraining/test/training_api/core/training_capi_tests.cc b/orttraining/orttraining/test/training_api/core/training_capi_tests.cc
index 9ae179faf4493..b1ed7c5e14666 100644
--- a/orttraining/orttraining/test/training_api/core/training_capi_tests.cc
+++ b/orttraining/orttraining/test/training_api/core/training_capi_tests.cc
@@ -423,7 +423,7 @@ TEST(TrainingCApiTest, UpdateParameterDifferentDevices) {
 TEST(TrainingCApiTest, ModuleAndOptimizerWithNominalState) {
   auto training_model_uri = MODEL_FOLDER "training_model.onnx";
   auto eval_model_uri = MODEL_FOLDER "eval_model.onnx";
-  auto optimizer_model_uri = MODEL_FOLDER "optimizer_model.onnx";
+  auto optimizer_model_uri = MODEL_FOLDER "adamw.onnx";
 
   Ort::Env env;
   Ort::SessionOptions session_options_for_complete_state;