Fixes incorrect focus during training