nit: I don't like this `return` here in the middle of the function. How about using a `else {` instead?