BUG: fix the tests on python 3 as well.