Update to onInjury method