fix if else error