FIX: Failing test for fit_transform()

mini-PR to fix a failing test