Allow passing in ZK client