Fix to CI error