A nit, but why not just return the value inside the if as well as here.