diff --git a/lib/generators/templates/migration.rb b/lib/generators/templates/migration.rb
index 0b39c01..751c95f 100644
--- a/lib/generators/templates/migration.rb
+++ b/lib/generators/templates/migration.rb
@@ -1,4 +1,4 @@
-class ActsAsFollowerMigration < ActiveRecord::Migration
+class ActsAsFollowerMigration < ActiveRecord::Migration[<%= Rails.version.to_f %>]
   def self.up
     create_table :follows, force: true do |t|
       t.references :followable, polymorphic: true, null: false