Merge pull request #172 from shuhaowu/patch-for-171

Fixed #171