Add test for code coverage