diff --git a/torchslim/quantizing/qat.py b/torchslim/quantizing/qat.py
index 4f3b944..4ff5652 100644
--- a/torchslim/quantizing/qat.py
+++ b/torchslim/quantizing/qat.py
@@ -1,4 +1,5 @@
 import sys
+import os
 
 import torch
 import copy