Modified test for CI