Fixing test for python3