fix bug
Merge pull request #10 from czheo/rjin

fix bug