fix in trainer