Fixing unittest import for python 2.6