Fix 'missing files from code coverage' issue