Update unit test cases