Fixes coverage on tests