improved code coverage and removed a warning