Update DNS test to new API