Use proper unittest import for py26 compat