Wrap the if in code.