Better handling of associations within polymorphic assocaitions.