[Feature] fix async unit test