Update the logging extensions to work with the new API changes.

Update the logging extensions to work with the new API changes.