Fix for test coverage