Correcting test to work with Python 3.x