fix code to support python 2.6