I didn't want to change the original code, I just moved it inside an `if` block.