Adjusted test to be Python 3 compliant