diff --git a/lib/will_paginate/deprecation.rb b/lib/will_paginate/deprecation.rb index 848ecf8f1..21b167c09 100644 --- a/lib/will_paginate/deprecation.rb +++ b/lib/will_paginate/deprecation.rb @@ -31,8 +31,8 @@ def []=(key, value) super end - def deprecate_key(*keys) - message = block_given? ? Proc.new : keys.pop + def deprecate_key(*keys, &block) + message = block_given? ? block : keys.pop Array(keys).each { |key| @deprecated[key] = message } end