Increase coverage

Sorry about that.  Accidentally removed this line during the refactor process but I added it back in now.