class ActiveRecord::Reflection::BelongsToReflection

def join_primary_key(klass = nil)
  polymorphic? ? association_primary_key(klass) : association_primary_key
end