diff --git a/invokeai/app/invocations/model.py b/invokeai/app/invocations/model.py
index 86936db93f..31d04cfd37 100644
--- a/invokeai/app/invocations/model.py
+++ b/invokeai/app/invocations/model.py
@@ -8,8 +8,8 @@ from .baseinvocation import (
     BaseInvocation,
     BaseInvocationOutput,
     FieldDescriptions,
-    InputField,
     Input,
+    InputField,
     InvocationContext,
     OutputField,
     UIType,
@@ -414,7 +414,7 @@ class SeamlessModeInvocation(BaseInvocation):
         default=None, description=FieldDescriptions.unet, input=Input.Connection, title="UNet"
     )
     vae: Optional[VaeField] = InputField(
-        default=None, description=FieldDescriptions.vae_model, input=Input.Any, title="VAE"
+        default=None, description=FieldDescriptions.vae_model, input=Input.Connection, title="VAE"
     )
     seamless_y: bool = InputField(default=True, input=Input.Any, description="Specify whether Y axis is seamless")
     seamless_x: bool = InputField(default=True, input=Input.Any, description="Specify whether X axis is seamless")