minor edit to __array_prepare__