Updating unit test for CI.