Change import to be able to use backported unittest module on python 2.6