diff --git a/update_setup_file.py b/update_setup_file.py
index 023a1bb..76cc01d 100644
--- a/update_setup_file.py
+++ b/update_setup_file.py
@@ -1,4 +1,5 @@
 import re
+import os
 
 
 version = os.environ['GITHUB_REF'].split('/')[-1]