diff --git a/nlp/cli.py b/nlp/cli.py
index f064365..8d06a36 100644
--- a/nlp/cli.py
+++ b/nlp/cli.py
@@ -12,8 +12,8 @@
 import requests
 from sklearn.feature_extraction.text import CountVectorizer
 from sklearn.metrics import f1_score, precision_score, recall_score
-from sklearn.naive_bayes import BernoulliNB, LogisticRegression
-
+from sklearn.naive_bayes import BernoulliNB
+from sklearn.linear_model import LogisticRegression
 from . import clf_path, config, config_path
 
 @click.group()