module ActiveRecord::Migration::Compatibility::V5_0::TableDefinition

def primary_key(name, type = :primary_key, **options)
  type = :integer if type == :primary_key
  super
end

def references(*args, **options)
  super(*args, type: :integer, **options)
end