FIX: don't force finite in imputation