diff --git a/bin/to_jit.py b/bin/to_jit.py
index f3b7cbb6..a96a7e79 100644
--- a/bin/to_jit.py
+++ b/bin/to_jit.py
@@ -1,5 +1,6 @@
 import os
 from pathlib import Path
+import sys
 
 import hydra
 import torch