fix api
Merge pull request #8 from Ch4s3/bug_fix

fix api