Improving test setup, adding a test to improve coverage.