Replace this block with a call to `init_loc_fields`.
Also, even if ` a.pidx` is 0, you want to check if `a.s` is defined, and if defined, set `A.s = reshape(a.s, dims)`