`grid_shape=grid_shape` here -- otherwise it's always `None`.