Add wait_test