Fix type error
Merge pull request #5 from rogerz/type-error

Accepted. Thanks :)