fix test for py3