Updating for test in CI