fix error
Merge pull request #1 from yongkangchen/patch-2

fix error