small fix to trainer ui