Fix bad scikit-learn import