Fix test on Python 3
Merge pull request #8 from mindflayer/master

Fix test on Python 3