Fix passing directory argument
Merge pull request #1 from alimony/patch-1

Fix passing directory argument