diff --git a/workers/comfyui-json/data_types.py b/workers/comfyui-json/data_types.py
index 9239ae5..d4cb7c6 100644
--- a/workers/comfyui-json/data_types.py
+++ b/workers/comfyui-json/data_types.py
@@ -6,6 +6,7 @@ from typing import Dict, Any
 from functools import cache
 from math import ceil
 from pathlib import Path
+import json
 
 from lib.data_types import ApiPayload, JsonDataException