Fix type error.
Merge pull request #2 from meteficha/fix-type-error

Fix type error.