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.