Tests now compatible with python 2.x and 3.x

Simple syntax change to make tests compatible with python3 as well as 2.