lib/stateful_enum/railtie.rb in stateful_enum-0.5.0 vs lib/stateful_enum/railtie.rb in stateful_enum-0.6.0

@@ -1,11 +1,15 @@
 # frozen_string_literal: true
 
 require 'stateful_enum/active_record_extension'
+require 'stateful_enum/state_inspection'
 
 module StatefulEnum
   class Railtie < ::Rails::Railtie
-    ActiveSupport.on_load :active_record do
-      ::ActiveRecord::Base.extend StatefulEnum::ActiveRecordEnumExtension
+    initializer 'stateful_enum' do
+      ActiveSupport.on_load :active_record do
+        ::ActiveRecord::Base.extend StatefulEnum::ActiveRecordEnumExtension
+        ::ActiveRecord::Base.include StatefulEnum::StateInspection
+      end
     end
   end
 end