fix if BUG