Fix test in Python 3
Merge pull request #5 from hemslo/feature/fix_test_in_py3

Fix test in Python 3