Fix missing require

public-activity · Jul 24, 2018 · 493b602 · 493b602
1 parent f190e3c
commit 493b602
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/lib/generators/public_activity/migration/templates/migration.rb b/lib/generators/public_activity/migration/templates/migration.rb
@@ -1,7 +1,5 @@
-require 'migrations_base.rb'
-
 # Migration responsible for creating a table with activities
-class CreateActivities < MigrationsBase
+class CreateActivities < (ActiveRecord.version.release() < Gem::Version.new('5.2.0') ? ActiveRecord::Migration : ActiveRecord::Migration[5.2])
   # Create table
   def self.up
     create_table :activities do |t|