fixed that yolo bug