trying to fix tests failing in python 3