fixed mock zk for tests