Fix broken test

Change-Id: I22f8dc58e4e1d6fdc357b31f6e63ec318c09d576