diff --git a/lib/will_paginate/deprecation.rb b/lib/will_paginate/deprecation.rb
index 2c44d1e60..d74b09a04 100644
--- a/lib/will_paginate/deprecation.rb
+++ b/lib/will_paginate/deprecation.rb
@@ -5,7 +5,7 @@ def self.debug() @debug; end
     def self.debug=(value) @debug = value; end
     self.debug = false
 
-    # Choose the default warn behavior according to RAILS_ENV.
+    # Choose the default warn behavior according to Rails.env.
     # Ignore deprecation warnings in production.
     BEHAVIORS = {
       'test'        => Proc.new { |message, callstack|
@@ -27,8 +27,8 @@ def self.warn(message, callstack = caller)
     end
 
     def self.default_behavior
-      if defined?(RAILS_ENV)
-        BEHAVIORS[RAILS_ENV.to_s]
+      if defined?(::Rails)
+        BEHAVIORS[::Rails.env.to_s]
       else
         BEHAVIORS['test']
       end