Fix helper method bug