Fix cts test due to API changes.

Related API changes at http://ag/526142

Change-Id: Iea76b72e2ea7b733677e2dd9feca264c68c49371
2 files changed