Fixed test on CI