diff --git a/.buildkite/pipeline.json.py b/.buildkite/pipeline.json.py
index c8178f3a1..126a6db35 100755
--- a/.buildkite/pipeline.json.py
+++ b/.buildkite/pipeline.json.py
@@ -17,6 +17,7 @@
 #
 
 import json
+import os
 
 from ml_pipeline import (
     step,