Fix if statement