diff --git a/examples/llm_finetuning/train_imdb_ray.py b/examples/llm_finetuning/train_imdb_ray.py
index 43dfc3ae0ef..2b055727af4 100644
--- a/examples/llm_finetuning/train_imdb_ray.py
+++ b/examples/llm_finetuning/train_imdb_ray.py
@@ -15,12 +15,15 @@
       type: auto_transformer
       pretrained_model_name_or_path: bigscience/bloom-3b
       trainable: true
-      adapter: lora
 
 output_features:
   - name: sentiment
     type: category
 
+
+adapter:
+  type: lora
+
 trainer:
   batch_size: 4
   epochs: 3