Set correct type
Merge pull request #17 from bvssvni/master

Set correct type