Custom tamed entities with delayed attacks attack other tamed entities

Steps to Reproduce:

1. Download the attached pack and create a world using it
2. Enable all experimental features available (I believe this isn't needed but I've always had experimental features on when testing)
3. Spawn the "Atlasaurus" entity
4. Tame it with wheat
5. Spawn a wolf
6. Tame the wolf
7. Make sure the dinosaur is standing and punch the wolf

Observed Behavior

The dinosaur will attack and kill the wolf even though they have been tamed by the same player.

Expected Behavior

The dinosaur will not attack the other entity.

Notes 

Removing the delayed attack component and using the melee attack component fixes this.

Edit: Included video to demonstrate that it doesn't happen every time but does happen sometimes