Reverted fix for Yolo