diff --git a/README.md b/README.md index d6d53e1..c4970aa 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 4dab44d..dd4d1b0 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)