Test updated to work with Python 2.6