Improved code
Merge pull request #197 from fabienrohrer/master

Minor code improvement