minor changes to get code coverage working