I should re-write the tests as proper unit tests with better coverage. Not a priority yet, just logging this for future me to consider.