Add test to fix coverage