Fix helper import