more strict when checking for IRB

hsume2 · Mar 14, 2011 · 05b003d · 05b003d
1 parent e657fb7
commit 05b003d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/hirb/view.rb b/lib/hirb/view.rb
@@ -178,7 +178,7 @@ def enable_output_method
         if defined? Ripl
           @output_method = true
           require 'ripl/hirb'
-        elsif defined? IRB
+        elsif defined? IRB::Irb
           @output_method = true
           ::IRB::Irb.class_eval do
             alias_method :non_hirb_view_output, :output_value
@@ -190,7 +190,7 @@ def output_value #:nodoc:
       end
 
       def disable_output_method
-        if defined?(IRB) && !defined? Ripl
+        if defined?(IRB::Irb) && !defined? Ripl
           ::IRB::Irb.send :alias_method, :output_value, :non_hirb_view_output
         end
         @output_method = nil