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