Fix build error with zookeeper