Update Companion for ai2-test.