Fix if statement slightly.