Fix the bug of judging
Merge pull request #41 from marshallshen/issue_36

Fix the bug of judging