make the unit test working