diff --git a/cnn_class2/use_pretrained_weights_vgg.py b/cnn_class2/use_pretrained_weights_vgg.py
index 01cbd619..bcad87e8 100644
--- a/cnn_class2/use_pretrained_weights_vgg.py
+++ b/cnn_class2/use_pretrained_weights_vgg.py
@@ -10,7 +10,7 @@
 from keras.applications.vgg16 import VGG16
 from keras.applications.vgg16 import preprocess_input
 from keras.preprocessing import image
-from keras.preprocessing.image import ImageDataGenerator
+from tensorflow.keras.preprocessing.image import ImageDataGenerator
 
 from sklearn.metrics import confusion_matrix
 import numpy as np
@@ -172,11 +172,11 @@ def get_confusion_matrix(data_path, N):
 plt.show()
 
 # accuracies
-plt.plot(r.history['acc'], label='train acc')
-plt.plot(r.history['val_acc'], label='val acc')
+plt.plot(r.history['accuracy'], label='train acc')
+plt.plot(r.history['val_accuracy'], label='val acc')
 plt.legend()
 plt.show()
 
 from util import plot_confusion_matrix
 plot_confusion_matrix(cm, labels, title='Train confusion matrix')
-plot_confusion_matrix(valid_cm, labels, title='Validation confusion matrix')
\ No newline at end of file
+plot_confusion_matrix(valid_cm, labels, title='Validation confusion matrix')