Fix function that was missing