Fix to work with Python 2.6

Only one simple change is needed, so might as well be compatible.