Whops.

Forgot to add in Affliction changes.