Corrected import of classes for Unittest.