fix import unittest for unittest2