diff --git a/iz_helpers/run.py b/iz_helpers/run.py
index bba2f9b..b838ff2 100644
--- a/iz_helpers/run.py
+++ b/iz_helpers/run.py
@@ -1,6 +1,6 @@
 import math, time, os
 import numpy as np
-from PIL import Image
+from PIL import Image, ImageFilter, ImageDraw
 from modules.ui import plaintext_to_html
 import modules.shared as shared
 from modules.processing import Processed, StableDiffusionProcessing