Minor mod to _src_align_and_flatten

This PR addresses minor issues in a helper function of your new regrid function.