Optimize getCommentedEntity()