Fix test

test success in python 2.6, and 2.7