Fix import error in pyhton 2.6.