Applying minor changes to add coverage.