fixing test cases with embedded zk integration