fix find wrong train problem