fix tests

Change-Id: I07c282f84c070d9001edd48ac2e90177b305a4a9