Fix error in code
Merge pull request #417 from WouterJ/patch-1

Fix error in code