Updating unit test cases