Add test to complete code coverage