diff --git a/lib/generators/redmine_plugin_model/templates/migration.rb b/lib/generators/redmine_plugin_model/templates/migration.rb
index f5026aebf..19f9302ab 100644
--- a/lib/generators/redmine_plugin_model/templates/migration.rb
+++ b/lib/generators/redmine_plugin_model/templates/migration.rb
@@ -1,4 +1,4 @@
-class <%= @migration_class_name %> < ActiveRecord::Migration
+class <%= @migration_class_name %> < ActiveRecord::Migration[<%= ActiveRecord::Migration.current_version %>]
   def change
     create_table :<%= @table_name %> do |t|
 <% attributes.each do |attribute| -%>