fix: zookeeper tests

Change-Id: Id96cd0d60f2536f238623155699b4dc49c56de47