[Fix] Add missing code in 8.18
Merge pull request #63 from zyuanhong/add_sth

[Fix] Add missing code in 8.18