Fix test

Change-Id: Ic1d8986a9b524eb0ce6d6b11ddffa9a54cee6515