diff --git a/Retrieval/plot_mrae_xaxis_k.py b/Retrieval/plot_mrae_xaxis_k.py
index 90e360b..9f6501a 100644
--- a/Retrieval/plot_mrae_xaxis_k.py
+++ b/Retrieval/plot_mrae_xaxis_k.py
@@ -89,6 +89,7 @@ for class_name in CLASS_NAMES:
             stds = np.asarray(stds)
 
             method_name = method_name.replace('NaiveQuery', 'Naive@$k$')
+            method_name = method_name.replace('KDEy-ML', 'KDEy')
             marker = next(markers)
             line = ax.plot(Ks, means, 'o-', label=method_name, color=None, linewidth=3, markersize=10, marker=marker)
             color = line[-1].get_color()
diff --git a/Retrieval/plot_mrae_xaxis_size.py b/Retrieval/plot_mrae_xaxis_size.py
index 0a362af..49ae9d8 100644
--- a/Retrieval/plot_mrae_xaxis_size.py
+++ b/Retrieval/plot_mrae_xaxis_size.py
@@ -54,6 +54,7 @@ for class_name in CLASS_NAMES:
 
             style = 'o-' if method_name != 'CC' else '--'
             method_name = method_name.replace('NaiveQuery', 'Naive@$k$')
+            method_name = method_name.replace('KDEy-ML', 'KDEy')
             marker=next(markers)
             line = ax.plot(X_DATA_SIZES, means, style, label=method_name, color=None, linewidth=3, markersize=10, marker=marker)
             color = line[-1].get_color()