fix support for scikit learn