training: fix