diff --git a/WAS_Node_Suite.py b/WAS_Node_Suite.py
index 8b63274..e2d319c 100644
--- a/WAS_Node_Suite.py
+++ b/WAS_Node_Suite.py
@@ -12961,7 +12961,7 @@ def INPUT_TYPES(cls):
 
     CATEGORY = "WAS Suite/Logic"
 
-    def vae_switch(self, vae_a, vae_b, boolean=True):
+    def vae_switch(self, vae_a, vae_b, boolean=True, **kwargs):
 
         if boolean:
             return (vae_a, )