fixes train route class assignment