Ensure logging is not shut down after upgrade

This looks like something is missing