Increase test coverage

Merge in when we get the test to become green