edited test code to get 100% coverage