just do that after the `if`