Fix tests

Change-Id: Ibcf4129efc1b21a91d0c27ea72f0a31ca9891e99