fix function return value