We are, so it should indeed be shape[0]