diff --git a/sdks/python/apache_beam/yaml/yaml_provider.py b/sdks/python/apache_beam/yaml/yaml_provider.py
index 3a88f6074b37..630e63c31d8a 100644
--- a/sdks/python/apache_beam/yaml/yaml_provider.py
+++ b/sdks/python/apache_beam/yaml/yaml_provider.py
@@ -169,8 +169,8 @@ def config_schema(self, type):
           self.schema_transforms()[self._urns[type]].configuration_schema)
 
   def requires_inputs(self, typ, args):
-    if self._urns[type] in self.schema_transforms():
-      return bool(self.schema_transforms()[self._urns[type]].inputs)
+    if self._urns[typ] in self.schema_transforms():
+      return bool(self.schema_transforms()[self._urns[typ]].inputs)
     else:
       return super().requires_inputs(typ, args)