fixing a bug for unit test on branch jz