Update essential test to use `wait_until` and remove sleep