Correct joblib usage