Place this after OnTakeDamage.