update no_color

1 files changed, 4 insertions, 4 deletions

diff --git a/lib/pry/helpers/colors.rb b/lib/pry/helpers/colors.rb
index 29f0ad1c..de5c2faf 100644
--- a/lib/pry/helpers/colors.rb
+++ b/lib/pry/helpers/colors.rb
@@ -77,11 +77,11 @@ module Pry::Helpers::Colors
   # Executes the block with `Pry.config.color` set to false.
   # @yield
   # @return [void]
-  def no_color(&block)
-    boolean = Pry.config.color
-    Pry.config.color = false
+  def no_color pry=(defined?(_pry_) && _pry_) || Pry, &block
+    boolean = pry.config.color
+    pry.config.color = false
     yield
   ensure
-    Pry.config.color = boolean
+    pry.config.color = boolean
   end
 end