diff --git a/fingpt/FinGPT_Benchmark/train_lora.py b/fingpt/FinGPT_Benchmark/train_lora.py
index f669fcba..79841f1c 100644
--- a/fingpt/FinGPT_Benchmark/train_lora.py
+++ b/fingpt/FinGPT_Benchmark/train_lora.py
@@ -23,7 +23,7 @@
     LoraConfig,
     get_peft_model,
     get_peft_model_state_dict,
-    prepare_model_for_int8_training,
+    prepare_model_for_kbit_training,
     set_peft_model_state_dict
 )
 from utils import *