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