Small fix

check the correct entity in the event.