Test Coverage Fix