Fixed Unit test