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

Fix return type incompatibility.