Fix helium test