use wait_until_assert instead of wait_until