YifanShenSZ bug fix