diff --git a/lib/generators/active_record/templates/create_impressions_table.rb b/lib/generators/active_record/templates/create_impressions_table.rb
index 13454b2..e0f2428 100644
--- a/lib/generators/active_record/templates/create_impressions_table.rb
+++ b/lib/generators/active_record/templates/create_impressions_table.rb
@@ -1,4 +1,4 @@
-class CreateImpressionsTable < ActiveRecord::Migration
+class CreateImpressionsTable < ActiveRecord::Migration<%= Rails::VERSION::MAJOR >= 5 ? "[#{Rails.version.to_f}]" : "" %>
   def self.up
     create_table :impressions, :force => true do |t|
       t.string :impressionable_type