if-else correction