Missing argument added.
Merge pull request #48 from 3Hren/fix

Missing argument added.