little fixes in training