FIX : broken fit_transform