Fix return type incompatibility.
Merge pull request #9 from dikmax/fix-return-type

Fix return type incompatibility.