Fix unit test during release