Add timeout to ZK connection