nit: I think here a `if/else` looks better :)