Add test to increase coverage