diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py
index 7f3706a04..4ac1f2238 100644
--- a/tests/basic/test_coder.py
+++ b/tests/basic/test_coder.py
@@ -1,3 +1,4 @@
+import os
 import tempfile
 import unittest
 from pathlib import Path