Optimize UT code by removing sleep logic.