Fix the function bug.