Fix non-contiguous reshape