Fix test in CI environment