Fixed test for CI