diff --git a/src/code.py b/src/code.py
index aada4e8..8059b07 100644
--- a/src/code.py
+++ b/src/code.py
@@ -9,6 +9,7 @@ import atexit
 import json
 import math
 import displayio
+import os
 import supervisor
 import sys
 import terminalio