diff --git a/computer_vision/cnn_classification.py b/computer_vision/cnn_classification.py
index b813b7103..115333eba 100644
--- a/computer_vision/cnn_classification.py
+++ b/computer_vision/cnn_classification.py
@@ -25,7 +25,7 @@ import numpy as np
 
 # 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