Fix soft impute model selection