Fix function error