diff --git a/huggingface_interface/train_lora.py b/huggingface_interface/train_lora.py
index 7556835..24847e5 100644
--- a/huggingface_interface/train_lora.py
+++ b/huggingface_interface/train_lora.py
@@ -4,7 +4,7 @@
 from datasets import Dataset
 from sacrebleu.metrics import BLEU, CHRF
 from peft import LoraConfig, get_peft_model
-from IndicTransToolkit import IndicProcessor, IndicDataCollator
+from IndicTransToolkit.processor import IndicProcessor, IndicDataCollator
 
 from transformers import (
     Seq2SeqTrainer,