zookeeper upgrade fix