Fix if test