fixed broken tests on python 3