Fix scikit model bug