Add test for uncovered code