~ Fixed missing return statement in YOLT