Fix  hei tests