Fix function name change