Modified so that the unit test will work even if we change the logging style in the future.