add ha upgrade support