diff --git a/1/model.py b/1/model.py
index 00270bf..b725d80 100644
--- a/1/model.py
+++ b/1/model.py
@@ -2,6 +2,7 @@ import triton_python_backend_utils as pb_utils
 import numpy as np
 import json
 import random
+import string
 
 class TritonPythonModel:
     def initialize(self, args):