diff --git a/subgen/subgen.py b/subgen/subgen.py
index 3ccff46..54f756a 100644
--- a/subgen/subgen.py
+++ b/subgen/subgen.py
@@ -9,6 +9,7 @@ import queue
 import logging
 import gc
 import io
+import random
 from typing import BinaryIO, Union, Any
 from fastapi import FastAPI, File, UploadFile, Query, Header, Body, Form, Request
 from fastapi.responses import StreamingResponse, RedirectResponse