fix hero train problem