Fix test that uses new API

Change-Id: Ib463dbece44ef1ec9842e3766c2c133b46ca63a5
1 file changed