py3: fix py2 tests