PMG - Fix issue in trainer mode