attempt to fix test for python 3