diff --git a/lib/table_migration.rb b/lib/table_migration.rb
index ca6de17..d1f8958 100644
--- a/lib/table_migration.rb
+++ b/lib/table_migration.rb
@@ -3,7 +3,7 @@
 
 migration_class =
   if ActiveRecord::VERSION::MAJOR >= 5
-    ActiveRecord::Migration[5.2]
+    ActiveRecord::Migration[ActiveRecord::VERSION::STRING.to_f]
   else
     ActiveRecord::Migration
   end
diff --git a/table_migrator.gemspec b/table_migrator.gemspec
index 197d99c..144eda0 100644
--- a/table_migrator.gemspec
+++ b/table_migrator.gemspec
@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name        = "table_migrator"
-  s.version     = '0.1.2'
+  s.version     = '0.1.3'
   s.platform    = Gem::Platform::RUBY
   s.authors     = ["Matt Freels", "Rohith Ravi", "Rick Olson",]
   s.email       = ["matt@freels.name"]