Fixed a test on python 3