HuaShen bug fix