Increased the test coverage