fix little bug in scikit-learn interface