if / else if correction