Fix for trains not updating. Again.