small fix to trainer ui
Merge pull request #44 from dmilisav/master

small fix to trainer ui