Fixed the unit test