Stupid if/then bug fix.