diff --git a/computer_vision/cnn_classification.py b/computer_vision/cnn_classification.py
index b813b71033f3..115333eba0d1 100644
--- a/computer_vision/cnn_classification.py
+++ b/computer_vision/cnn_classification.py
@@ -25,7 +25,7 @@
 
 # Importing the Keras libraries and packages
 import tensorflow as tf
-from tensorflow.keras import layers, models
+from keras import layers, models
 
 if __name__ == "__main__":
     # Initialising the CNN