replace with `return`