diff --git a/README.md b/README.md index 5cdac58..c7be319 100644 --- a/README.md +++ b/README.md @@ -73,9 +73,9 @@ def call(env) if whitelisted?(req) @app.call(env) elsif blacklisted?(req) - blacklisted_response[env] + self.class.blacklisted_response.call(env) elsif throttled?(req) - throttled_response[env] + self.class.throttled_response.call(env) else tracked?(req) @app.call(env) diff --git a/lib/rack/attack.rb b/lib/rack/attack.rb index c98f066..d8308a5 100644 --- a/lib/rack/attack.rb +++ b/lib/rack/attack.rb @@ -96,9 +96,9 @@ class Rack::Attack if whitelisted?(req) @app.call(env) elsif blacklisted?(req) - self.class.blacklisted_response[env] + self.class.blacklisted_response.call(env) elsif throttled?(req) - self.class.throttled_response[env] + self.class.throttled_response.call(env) else tracked?(req) @app.call(env)