Fixed Unit Test setup.