Fixed Unit test error.