diff --git a/scripts/dlss_ext.py b/scripts/dlss_ext.py
index 73d3513a6..f7a4d7580 100644
--- a/scripts/dlss_ext.py
+++ b/scripts/dlss_ext.py
@@ -392,7 +392,7 @@ class DLSSScript(scripts_manager.Script):
 
     def postprocess(self, p: processing.StableDiffusionProcessing, pp: processing.Processed, *args, **kwargs): # pylint: disable=arguments-differ,unused-argument
         if p.xyz: # do not postprocessing when running in xyz mode
-            return
+            return pp
         _pp = dlss(p, pp, *args, **kwargs)
         # postprocess triggers after initial images have already been saved
         if _pp is not None and hasattr(_pp, 'images') and _pp.images is not None: