diff --git a/torch_xla/core/xla_model.py b/torch_xla/core/xla_model.py
index 8f28ab8a499..cdc4d84195b 100644
--- a/torch_xla/core/xla_model.py
+++ b/torch_xla/core/xla_model.py
@@ -18,6 +18,7 @@
 import torch_xla.debug.metrics_saver as ms
 import torch_xla.utils.utils as xu
 import torch_xla.utils.closures as xc
+import os
 
 _DEVICES = xu.LazyProperty(lambda: torch_xla._XLAC._xla_get_devices())