Just ran tests, replace dwi_slices by vol_idx