Add extra test for 100% code coverage