diff --git a/Code/Install_Packages.txt b/Code/Install_Packages.txt
index 65a75a7176..7b613bb011 100644
--- a/Code/Install_Packages.txt
+++ b/Code/Install_Packages.txt
@@ -1,7 +1,6 @@
 h5py
 numpy
 scikit-learn
-sklearn
 keras
 opencv-python
 pyttsx3
diff --git a/Code/cnn_model_train.py b/Code/cnn_model_train.py
index 00473bde75..0cb168b361 100644
--- a/Code/cnn_model_train.py
+++ b/Code/cnn_model_train.py
@@ -7,7 +7,7 @@
 from keras.layers import Dense
 from keras.layers import Dropout
 from keras.layers import Flatten
-from keras.layers.convolutional import Conv2D
+from tensorflow.keras.layers import Conv2D
 from keras.layers.convolutional import MaxPooling2D
 from keras.utils import np_utils
 from keras.callbacks import ModelCheckpoint