lib/combustion/database/migrate.rb in combustion-1.1.0 vs lib/combustion/database/migrate.rb in combustion-1.1.1

@@ -42,10 +42,16 @@
   def engine_paths_exist_in?(paths)
     paths.include?(engine_path.join("db/migrate").to_s)
   end
 
   def migration_context
-    ActiveRecord::MigrationContext.new paths
+    if ActiveRecord::MigrationContext.instance_method(:initialize).arity <= 1
+      ActiveRecord::MigrationContext.new paths
+    else
+      ActiveRecord::MigrationContext.new(
+        paths, ActiveRecord::Base.connection.schema_migration
+      )
+    end
   end
 
   def migrator
     @migrator ||= ActiveRecord::Migrator
   end