diff --git a/lib/mini_record/auto_schema.rb b/lib/mini_record/auto_schema.rb
index cd8cdea..9fd2612 100644
--- a/lib/mini_record/auto_schema.rb
+++ b/lib/mini_record/auto_schema.rb
@@ -18,6 +18,12 @@ def init_table_definition(connection)
         when -5
           # Rails 4.1
           ActiveRecord::ConnectionAdapters::TableDefinition.new(connection.native_database_types, table_name, false, {}, nil)
+        when -4
+          # Rails 5 beta
+          ActiveRecord::ConnectionAdapters::TableDefinition.new(table_name, false, {}, nil)
+        when -2
+          # Rails 5
+          ActiveRecord::ConnectionAdapters::TableDefinition.new(table_name)
         else
           raise ArgumentError,
             "Unsupported number of args for ActiveRecord::ConnectionAdapters::TableDefinition.new()"