fix: work on the train.