diff --git a/dga/lstm.py b/dga/lstm.py
index acad8f2..a7c87a4 100644
--- a/dga/lstm.py
+++ b/dga/lstm.py
@@ -12,7 +12,9 @@
 from sklearn import preprocessing
 from sklearn.utils import shuffle
 from sklearn.svm import SVC
-from sklearn import cross_validation
+
+#from sklearn import cross_validation
+from sklearn import model_selection
 from keras.preprocessing import sequence
 from keras.models import Sequential
 from keras.layers import Dense,Embedding