The if statement below seems to handle this.