Fix 100% code coverage