`arr.shape` is the global shape here right? Shouldn't this be `local_shape`?