Fix test for python3