[YOLO-LIFEHACK] - proper entity element in output