Fix race condition in UT cases