diff --git a/pytorch3d/implicitron/tools/stats.py b/pytorch3d/implicitron/tools/stats.py
index 32e7d703..a2d473e2 100644
--- a/pytorch3d/implicitron/tools/stats.py
+++ b/pytorch3d/implicitron/tools/stats.py
@@ -434,12 +434,9 @@ class Stats(object):
                 plt.gca().yaxis.label.set_color(c[0:3] * 0.75)
                 plt.legend(tmodes)
                 gcolor = np.array(mcolors.to_rgba("lightgray"))
-                plt.grid(
-                    b=True, which="major", color=gcolor, linestyle="-", linewidth=0.4
-                )
-                plt.grid(
-                    b=True, which="minor", color=gcolor, linestyle="--", linewidth=0.2
-                )
+                grid_params = {"visible": True, "color": gcolor}
+                plt.grid(**grid_params, which="major", linestyle="-", linewidth=0.4)
+                plt.grid(**grid_params, which="minor", linestyle="--", linewidth=0.2)
                 plt.minorticks_on()
 
             plt.tight_layout()