Minor API update for test.