Modify test to not use joblib