fix the bug
Merge pull request #2 from alex-phn/bug001

fix the bug