Quick fix to handle Rails 4 associations.