modify the code
Merge pull request #335 from youyong205/master

modify the code