Fix wrong argument type.
Merge pull request #27 from nschum/fix-argument-type

Fix wrong argument type