bug fix for mean_impute