fix small return error