Fix unit test callback arguments.