fix tests

Change-Id: I1839e9b1e8dc5576322c5ce88d97fdbd01c67de4