(#405) - Fix tests when running coverage