Nit: you can move the `if` statement inside the `context.become(...)`