Fix issue with undefined variable
Merge pull request #3 from rtripault/fixes

Fix issue with undefined variable