remove unnecessary wait

[3r1ccc]

Signed-off-by: Eric Chang ericcc@infuseai.io

PR checklist

• [X] Ensure you have added or ran the appropriate tests for your PR.
• [X] DCO signed

What type of PR is this? Chore

What this PR does / why we need it: To shorten running time

Which issue(s) this PR fixes: Remove unnecessary wait

Special notes for your reviewer:

Does this PR introduce a user-facing change?:

NONE