diff --git a/extensions/dns/model.rb b/extensions/dns/model.rb
index 2a03f2fc8..84ac3555a 100644
--- a/extensions/dns/model.rb
+++ b/extensions/dns/model.rb
@@ -14,9 +14,10 @@ module DNS
 
     storage_names[:default] = 'extension_dns_rules'
 
-    property :id, Serial
-    property :pattern, Object
-    property :block, Text
+    property :id, Serial       # Unique identifier
+    property :pattern, Object  # Query pattern
+    property :type, Object     # Resource type
+    property :block, Text      # Associated callback
 
   end
 
diff --git a/extensions/dns/ruby/rubydns.rb b/extensions/dns/ruby/rubydns.rb
index 3dbc837e8..fb5e21c62 100644
--- a/extensions/dns/ruby/rubydns.rb
+++ b/extensions/dns/ruby/rubydns.rb
@@ -55,7 +55,8 @@ module RubyDNS
         # Add new rule to database
         BeEF::Core::Models::DNS::Rule.create(
           :id => id,
-          :pattern => pattern,
+          :pattern => pattern[0],
+          :type => pattern[1],
           :block => block.to_source
         )
       end