lib/ridgepole/migration_ext.rb in ridgepole-0.6.4.beta6 vs lib/ridgepole/migration_ext.rb in ridgepole-0.6.4.beta7

@@ -1,10 +1,10 @@
 require 'active_record/migration'
 
 module Ridgepole
   module MigrationExt
-    def self.included(klass)
+    def self.prepended(klass)
       klass.class_eval do
         cattr_accessor :time_recorder
         cattr_accessor :disable_logging
       end
     end
@@ -62,8 +62,8 @@
   end
 end
 
 module ActiveRecord
   class Migration
-    include Ridgepole::MigrationExt
+    prepend Ridgepole::MigrationExt
   end
 end