Fix function refs