TST: fix test on python3