diff --git a/script_examples/websockets_api_example.py b/script_examples/websockets_api_example.py
index 62afc869c..c4da51b75 100644
--- a/script_examples/websockets_api_example.py
+++ b/script_examples/websockets_api_example.py
@@ -7,6 +7,7 @@ import json
 import urllib.request
 import urllib.parse
 
+
 server_address = "127.0.0.1:8188"
 client_id = str(uuid.uuid4())
 
@@ -153,11 +154,10 @@ ws.connect("ws://{}/ws?clientId={}".format(server_address, client_id))
 images = get_images(ws, prompt)
 
 #Commented out code to display the output images:
-
+# from PIL import Image
+# import io
 # for node_id in images:
 #     for image_data in images[node_id]:
-#         from PIL import Image
-#         import io
 #         image = Image.open(io.BytesIO(image_data))
 #         image.show()