diff --git a/lib/validates_timeliness/railtie.rb b/lib/validates_timeliness/railtie.rb index de13df6..6b4bb26 100644 --- a/lib/validates_timeliness/railtie.rb +++ b/lib/validates_timeliness/railtie.rb @@ -1,7 +1,10 @@ module ValidatesTimeliness class Railtie < Rails::Railtie initializer "validates_timeliness.initialize_active_record", :after => 'active_record.initialize_timezone' do - ValidatesTimeliness.default_timezone = ActiveRecord::Base.default_timezone + ActiveSupport.on_load(:active_record) do + ValidatesTimeliness.default_timezone = ActiveRecord::Base.default_timezone + ValidatesTimeliness.extend_orms = [ :active_record ] + end end initializer "validates_timeliness.initialize_restriction_errors" do @@ -9,7 +12,3 @@ module ValidatesTimeliness end end end - -ActiveSupport.on_load(:active_record) do - ValidatesTimeliness.extend_orms = [ :active_record ] -end