diff --git a/nodes.py b/nodes.py
index 549ebb0..2004c25 100644
--- a/nodes.py
+++ b/nodes.py
@@ -5,7 +5,7 @@ class BridgeStoreNode:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "data": ("ANY",),
+                "data": (["LATENT", "CONDITIONING", "IMAGE", "STRING", "ANY"],),
                 "bridge_id": ("STRING", {"default": "my_key"})
             }
         }