fixed test for python 3