Tests are working with python 3