fix function return type