Fixed old bug with train updates