clean up yolo warning a bit