diff --git a/gridcorner.py b/gridcorner.py
index 304f0a2179f1cdd8460d43635b14a090d23451ef..0b71acb351e7b14c691bcfc8a015823929c2f45a 100644
--- a/gridcorner.py
+++ b/gridcorner.py
@@ -56,6 +56,7 @@ def _get_fig_and_axes(ndim, factor, whspace):
     plotdim = factor * ndim + factor * (ndim - 1.) * whspace
     dim = lbdim + plotdim + trdim
     fig, axes = plt.subplots(ndim, ndim, figsize=(dim, dim))
+    axes = np.atleast_2d(axes) # allow single-parameter plots
 
     # Format the figure.
     lb = lbdim / dim