Changed component to entity listener