correct the function class