remove yolo and make it stable