Fix if bug