Fix CTS test for updated API

Change-Id: Id77301c8752da3cf25cc1f8073471c2c5330c4af
2 files changed