Add test for coverage