Get back to 100% test coverage

Missed some helper functions in the test coverage. Updated it so that
100% of the lines are covered now