diff --git a/py/easyNodes.py b/py/easyNodes.py
index 8046dfe..627b389 100644
--- a/py/easyNodes.py
+++ b/py/easyNodes.py
@@ -412,6 +412,7 @@ class easySampler:
         return (
             pipe,
             pipe.get("images"),
+            pipe.get("model"),
             pipe.get("positive"),
             pipe.get("negative"),
             pipe.get("samples"),