small optimization
Merge pull request #5 from DaanHoogland/master

small optimization