found major bug in training