fix tests

Change-Id: I97d769c76f23cd259f92db6e5b27735f4cf66927