lib/generators/audited/upgrade_generator.rb in audited-4.3.0 vs lib/generators/audited/upgrade_generator.rb in audited-4.4.0

@@ -1,14 +1,16 @@
 require 'rails/generators'
 require 'rails/generators/migration'
 require 'active_record'
 require 'rails/generators/active_record'
 require 'generators/audited/migration'
+require 'generators/audited/migration_helper'
 
 module Audited
   module Generators
     class UpgradeGenerator < Rails::Generators::Base
       include Rails::Generators::Migration
+      include Audited::Generators::MigrationHelper
       extend Audited::Generators::Migration
 
       source_root File.expand_path("../templates", __FILE__)
 
       def copy_templates