diff --git a/common/logging_extra.py b/common/logging_extra.py
index 64da0d0de..b5d07f220 100644
--- a/common/logging_extra.py
+++ b/common/logging_extra.py
@@ -1,9 +1,11 @@
+import io
 import os
 import sys
 import copy
 import json
 import socket
 import logging
+import traceback
 from threading import local
 from collections import OrderedDict
 from contextlib import contextmanager