Add two more tests for 100% coverage