diff --git a/EMRI_DET/validate.py b/EMRI_DET/validate.py
index a6476ef05cfee25b8f37d78cfb36c2ccd64db5f4..bafcbb2e46c4c95c729708405cd8ab12988a2d66 100644
--- a/EMRI_DET/validate.py
+++ b/EMRI_DET/validate.py
@@ -228,7 +228,7 @@ def grid_heatmap_corner(dataframe, truth_column, pred_column, log=True, ratio=Fa
             ax[j,i].grid(False)
             if i != 0:
                 ax[j,i].yaxis.set_visible(False)
-            if j != 4:
+            if j != nparams-1:
                 ax[j,i].xaxis.set_visible(False)
 
             num += 1