PY3 Fix test for Python 3