Add first_wait test