BUG: Remove unused imports and unittest2 dependency.