diff --git a/src/skit/ModelMetrics.py b/src/skit/ModelMetrics.py
index f60ac4e..8d71f17 100644
--- a/src/skit/ModelMetrics.py
+++ b/src/skit/ModelMetrics.py
@@ -52,7 +52,7 @@ def val_metric_key(self):
         if self == Metric.ACCURACY:
             return "val_accuracy"
         elif self == Metric.AUC:
-            return "val_auc"
+            return "auc"
         elif self == Metric.VAL_AUC:
             return "val_auc"
         elif self == Metric.VAL_ACCURACY:
@@ -76,7 +76,7 @@ def plot_labels(self):
             }
         elif self == Metric.AUC or self == Metric.VAL_AUC:
             return {
-                'Accuracy': {
+                'AUC': {
                     'Training AUC': 'auc',
                     'Validation AUC': 'val_auc'
                 },