From ccf06da461d7b16196ca263126513763ad566d2a Mon Sep 17 00:00:00 2001
From: Ben Atkins <benatkins@fullbridge.com>
Date: Mon, 19 Aug 2013 12:02:11 -0400
Subject: [PATCH] Leveraging ActiveRecord::Generators::Base class for
 implementation of the :next_migration_number class method on the
 InstallGenerator

---
 lib/generators/paper_trail/install_generator.rb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/generators/paper_trail/install_generator.rb b/lib/generators/paper_trail/install_generator.rb
index 85ccb69f..af7a970b 100644
--- a/lib/generators/paper_trail/install_generator.rb
+++ b/lib/generators/paper_trail/install_generator.rb
@@ -1,5 +1,6 @@
 require 'rails/generators'
 require 'rails/generators/migration'
+require 'rails/generators/active_record'
 
 module PaperTrail
   class InstallGenerator < ::Rails::Generators::Base
@@ -16,8 +17,7 @@ module PaperTrail
     end
 
     def self.next_migration_number(dirname)
-      next_migration_number = current_migration_number(dirname) + 1
-      ActiveRecord::Migration.next_migration_number(next_migration_number)
+      ActiveRecord::Generators::Base.next_migration_number(dirname)
     end
   end
 end