Add test for coverage 100%