Fixed Unit test.