PY3 Fix test on Python 3