Updating test as per API changes

Change-Id: I746bf48254ebfcad18426d47c61a54f8a11f5cf8
1 file changed