lib/thor/monkies/shell.rb in berkshelf-1.4.6 vs lib/thor/monkies/shell.rb in berkshelf-2.0.0.beta

@@ -1,8 +1,5 @@
 require 'berkshelf/ui'
 
 # @author Seth Vargo <[e-mail]>
-class ::Thor::Shell::Color
-  # Include the Berkshelf UI methods - this is used by both
-  # Vagrant and Berkshelf UI
-  include ::Berkshelf::UI
-end
+# Include the Berkshelf UI methods - this is used by both Vagrant and Berkshelf
+Thor::Base.shell.send(:include, Berkshelf::UI)