Fix for trains not updating.