Fixed test that wasn't Python3 compatible