Improve DNS testing