fix trainer class in yellow bug