lib/pry-theme.rb in pry-theme-1.1.1 vs lib/pry-theme.rb in pry-theme-1.1.2

@@ -14,10 +14,13 @@
 require 'pry-theme/color'
 require 'pry-theme/color_table'
 require 'pry-theme/basic_editor'
 require 'pry-theme/commands'
 
+Pry.config.hooks
+  .add_hook(:when_started, :pry_theme, PryTheme::WhenStartedHook.new)
+
 module PryTheme
 
   # The VERSION file must be in the root directory of the library.
   VERSION_FILE = File.expand_path('../../VERSION', __FILE__)
 
@@ -94,8 +97,6 @@
     # @return [Class] the class, which corresponds to the given +color+
     def color_const(color)
       const_get(:"Color#{ color }")
     end
   end
-
-  Pry.config.hooks.add_hook(:when_started, :pry_theme, WhenStartedHook.new)
 end