diff --git a/kernel_ridge_linear_model/kernel_ridge.py b/kernel_ridge_linear_model/kernel_ridge.py
index 19d5c6da4d2b423530aeaadba419ac0cf58d1b51..fd1f277be1f9051ec3fb3bbb7dc533dc0715ba42 100755
--- a/kernel_ridge_linear_model/kernel_ridge.py
+++ b/kernel_ridge_linear_model/kernel_ridge.py
@@ -274,6 +274,8 @@ if (generate_plots):
         kernel_ridge_ax.set_xlabel('Configuration')
         kernel_ridge_ax.set_ylabel(r'Energy consumed/Workload ($\times 10E-11$)')
         kernel_ridge_ax.tick_params(size=8)
+        _ = kernel_ridge_ax.set_title("All datas")
+
     else : 
         
         kernel_ridge_ax.bar(X_user_friendly_test,predicted_y_test, width=0.4)
@@ -282,8 +284,8 @@ if (generate_plots):
         kernel_ridge_ax.set_xlabel('Configuration')
         kernel_ridge_ax.set_ylabel(r'Energy consumed/Workload ($\times 10E-11$)')
         kernel_ridge_ax.tick_params(size=8)
+        _ = kernel_ridge_ax.set_title("Predicted data\n using kernel ridge, R2 = " + str(utils.compute_r2_score(y_test, predicted_y_test)))
 
-    _ = kernel_ridge_ax.set_title("Predicted data\n using kernel ridge, R2 = " + str(utils.compute_r2_score(y_test, predicted_y_test)))
     R2_score = utils.compute_r2_score(y_test, predicted_y_test)
     print(" R2 error = ", utils.compute_r2_score(y_test, predicted_y_test))
     if(search_ridge_coeff):