fix refactoring bug which changed everything to _single_fit_transform