fix for soft impute