diff --git a/comfy_api_nodes/nodes_recraft.py b/comfy_api_nodes/nodes_recraft.py
index e369c4b7e..8cb16cc4e 100644
--- a/comfy_api_nodes/nodes_recraft.py
+++ b/comfy_api_nodes/nodes_recraft.py
@@ -32,6 +32,7 @@ from comfy_api_nodes.apinode_utils import (
 )
 from server import PromptServer
 
+import os
 import torch
 from io import BytesIO
 from PIL import UnidentifiedImageError