I attempted to add it in pre-attributes, but due to a bukkit limitation, it was not possible at the time. It should be now, as they've fixed the bug with the wither damage, however, it needs to be handled the same way we handle fire ticks. Even our firetick system is not perfect though, and often steals kills away from the players. I can't say we'll be able to fix this 100%.