Fix if checking