diff --git a/sdks/python/apache_beam/yaml/integration_tests.py b/sdks/python/apache_beam/yaml/integration_tests.py
index a69e0b6ba4e1..72b3918195da 100644
--- a/sdks/python/apache_beam/yaml/integration_tests.py
+++ b/sdks/python/apache_beam/yaml/integration_tests.py
@@ -191,7 +191,7 @@ def parse_test_files(filepattern):
 
 
 logging.getLogger().setLevel(logging.INFO)
-parse_test_files(os.path.join(os.path.dirname(__file__), 'tests', 'map.yaml'))
+parse_test_files(os.path.join(os.path.dirname(__file__), 'tests', '*.yaml'))
 
 if __name__ == '__main__':
   logging.getLogger().setLevel(logging.INFO)