diff --git a/test/unit/extensions/tc_dns.rb b/test/unit/extensions/tc_dns.rb
index bfb7ca2c3..af0ca9e42 100644
--- a/test/unit/extensions/tc_dns.rb
+++ b/test/unit/extensions/tc_dns.rb
@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2013 Wade Alcorn - wade@bindshell.net
+# Copyright (c) 2006-2014 Wade Alcorn - wade@bindshell.net
 # Browser Exploitation Framework (BeEF) - http://beefproject.com
 # See the file 'doc/COPYING' for copying permission
 #
@@ -39,8 +39,10 @@ class TC_Dns < Test::Unit::TestCase
 
   # Checks for required settings in config file
   def test_02_config
+    assert(@@dns_config.has_key?('protocol'))
     assert(@@dns_config.has_key?('address'))
     assert(@@dns_config.has_key?('port'))
+    assert(@@dns_config.has_key?('upstream'))
   end
 
   # Verifies public interface