fix soft impute