Fix the bug about function return value.