Added test for 100% coverage