Modify test cases to get 100% code coverage