Unit test case bug fixed