BUG: Initialize shape/index in nditer to fix heisenbug