diff --git a/main.py b/main.py
index d244ebd..80500dc 100644
--- a/main.py
+++ b/main.py
@@ -5,7 +5,7 @@ import io
 from typing import Union, Optional
 from fastapi import FastAPI, File, UploadFile
 from pydantic import BaseModel
-from starlette.responses import FileResponse
+from starlette.responses import FileResponse, JSONResponse
 from fastapi import FastAPI, Response, status
 import pytjnetwork as tj
 from tjnetwork import *