Add test case for coverage