Update unit test files for base classes