fix train vanish bug