Fix test on CI