Fix function rename bug