Fixing zkclient2 tests.