Restored the logging for tests