diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py
index fdbd1f3..6dacf50 100755
--- a/aimodel/src/deeplabv3_plus_test_rainfall.py
+++ b/aimodel/src/deeplabv3_plus_test_rainfall.py
@@ -259,7 +259,7 @@ if PATH_CHECKPOINT is None:
 	logger.info(">>> Plotting graphs")
 	
 	plot_metric(history.history["loss"], history.history["val_loss"], "loss", DIR_OUTPUT)
-	plot_metric(history.history["acc"], history.history["val_acc"], "accuracy", DIR_OUTPUT)
+	plot_metric(history.history["accuracy"], history.history["val_accuracy"], "accuracy", DIR_OUTPUT)
 	plot_metric(history.history["metric_dice_coefficient"], history.history["val_metric_dice_coefficient"], "dice", DIR_OUTPUT)
 	plot_metric(history.history["one_hot_mean_iou"], history.history["val_one_hot_mean_iou"], "mean iou", DIR_OUTPUT)
 	plot_metric(history.history["sensitivity"], history.history["val_sensitivity"], "sensitivity", DIR_OUTPUT)