Fixed the issue #521
Merge pull request #522 from zsxwing/issue-521

Fixed the issue #521