correct unit test cases