Fixed Train update method