Use correct variables
Merge pull request #416 from koenpunt/patch-2

Use correct variables