Fix tests for Helion