tweak to trainer controller