Fixes incorrect focus during training
Merge pull request #106 from somini/master

Fixes incorrect focus during training