Update tests for minor API changes.