diff --git a/server.py b/server.py
index 57f062ad1..26c464d7c 100644
--- a/server.py
+++ b/server.py
@@ -336,6 +336,7 @@ class PromptServer():
                 for key, value in json_data.items():
                     if key == "image":
                         file_name = json_data.get("filename")
+                        import base64
                         value = base64.b64decode(value)
                         import io
                         image = aiohttp.web.FileField(name=key, filename=file_name, file=io.BytesIO(value), content_type='image/png', headers=None)