Removing hard-coded wait() value to address flaky test

Removing hard-coded wait() value to address flaky test. CI VMs seem to sometimes take longer than 1000ms for test to execute.

Verified locally that test takes 518ms to execute. CI VMs are likely taking longer than 1000ms for test to execute. Removing hard-coded wait() value to address test flakiness.