bug fix yolo