Add test for full test coverage