diff --git a/language/llama2-70b/SUT_API.py b/language/llama2-70b/SUT_API.py
index 0b1ebd0c98..6331e33e07 100644
--- a/language/llama2-70b/SUT_API.py
+++ b/language/llama2-70b/SUT_API.py
@@ -1,4 +1,5 @@
 import os
+import sys
 import time
 import numpy as np
 import array