Backport HA changes