Fixed incomplete test coverage