test/dummy/config/initializers/migration_class.rb in devise-security-0.13.0 vs test/dummy/config/initializers/migration_class.rb in devise-security-0.14.0.rc1

@@ -1,8 +1,10 @@
 # frozen_string_literal: true
 
-MIGRATION_CLASS =
-  if ActiveRecord::VERSION::MAJOR >= 5
-    ActiveRecord::Migration[4.2]
-  else
-    ActiveRecord::Migration
-  end
+if DEVISE_ORM == :active_record
+  MIGRATION_CLASS =
+    if Rails.gem_version >= Gem::Version.new('5.0')
+      ActiveRecord::Migration[Rails.version.to_f]
+    else
+      ActiveRecord::Migration
+    end
+end