fix the return type