minor pythonic fix
Merge pull request #4 from kovacsbalu/master

minor pythonic fix