diff --git a/lib/config.py b/lib/config.py
index 212c323..21929c8 100644
--- a/lib/config.py
+++ b/lib/config.py
@@ -19,6 +19,7 @@
 
 from absl import logging
 
+import io
 
 
 class _ConfigManager:
diff --git a/lib/utils.py b/lib/utils.py
index 611e891..0268b8e 100644
--- a/lib/utils.py
+++ b/lib/utils.py
@@ -35,6 +35,7 @@
 
 from absl import logging
 
+import io
 from lib import constants