diff --git a/lib/awesome_print/ext/sequel.rb b/lib/awesome_print/ext/sequel.rb
index 8ee8b90..b402471 100644
--- a/lib/awesome_print/ext/sequel.rb
+++ b/lib/awesome_print/ext/sequel.rb
@@ -46,7 +46,7 @@ module AwesomePrint
 
     # Format Sequel Model class.
     #------------------------------------------------------------------------------
-    def awesome_sequel_model_class
+    def awesome_sequel_model_class(object)
       data = object.db_schema.inject({}) {|h, (name,data)| h.merge(name => data[:db_type])}
       "class #{object} < #{object.superclass} " << awesome_hash(data)
     end