Get test coverage working