diff --git a/nlp/cli.py b/nlp/cli.py
index 1224dc9..639bc5b 100644
--- a/nlp/cli.py
+++ b/nlp/cli.py
@@ -20,7 +20,7 @@
 import tensorflow as tf
 import keras as keras
 from tensorflow.keras.models import load_model
-from tensorflow.keras.preprocessing.text import text
+from keras.preprocessing.text import text
 from tensorflow.keras.preprocessing.sequence import pad_sequences
 from sklearn.preprocessing import LabelEncoder