MAINT: fix one more occurrence of np.prod inside reshape.