Allays assigned to a player can be set on fire with sweeping and fire aspect

When attacking a mob with a fire aspect sword the sweeping affect can hit an assigned allay within range and light the allay on fire.

What I expected to happen was...:
The allay should be immune to damage from the player who owns it. This should prevent the allay from being affected by fire aspect.

What actually happened was...:
When attacking a mob with the assigned allay nearby the allay does not appear to take damage from the sweeping affect but is still lit on fire.

Steps to Reproduce:
1. Have an allay assigned to a player (give it an item).
2. Have a fire aspect sword.
3. Attack a mob with the fire aspect sword while the allay is in range of the sweeping effect.