fix the return value error