diff --git a/kairon/nlu/classifiers/llm.py b/kairon/nlu/classifiers/llm.py
index c1906408c..a383d03f5 100644
--- a/kairon/nlu/classifiers/llm.py
+++ b/kairon/nlu/classifiers/llm.py
@@ -16,7 +16,7 @@
 from rasa.engine.storage.storage import ModelStorage
 from rasa.nlu.classifiers.classifier import IntentClassifier
 from rasa.nlu.extractors.extractor import EntityExtractorMixin
-from rasa.shared.nlu.constants import TEXT, INTENT, ENTITIES, ENTITY_ATTRIBUTE_TYPE
+from rasa.shared.nlu.constants import TEXT, INTENT, ENTITIES, ENTITY_ATTRIBUTE_TYPE, EXTRACTOR
 from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.utils.io import create_directory_for_file