diff --git a/scripts/dynamic_thresholding.py b/scripts/dynamic_thresholding.py
index 60a96b2..a5d40d2 100644
--- a/scripts/dynamic_thresholding.py
+++ b/scripts/dynamic_thresholding.py
@@ -234,7 +234,7 @@ def combine_denoised(self, x_out, conds_list, uncond, cond_scale):
 ######################### XYZ Plot Script Support logic #########################
 
 def make_axis_options():
-    xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ == "xyz_grid.py"][0].module
+    xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ in ("xyz_grid.py", "scripts.xyz_grid")][0].module
     def apply_mimic_scale(p, x, xs):
         if x != 0:
             setattr(p, "dynthres_enabled", True)