diff --git a/rules/python/third_parties/google_dataflow.yml b/rules/python/third_parties/google_dataflow.yml
index c02b6579..be906dc3 100644
--- a/rules/python/third_parties/google_dataflow.yml
+++ b/rules/python/third_parties/google_dataflow.yml
@@ -1,8 +1,17 @@
 imports:
   - python_shared_lang_datatype
+  - python_shared_lang_import1
 patterns:
-  - pattern: beam.Create($<...>$<DATA_TYPE>$<...>)
+  - pattern: $<CREATE>($<...>$<DATA_TYPE>$<...>)
     filters:
+      - variable: CREATE
+        detection: python_shared_lang_import1
+        scope: cursor
+        filters:
+          - variable: MODULE1
+            values: [apache_beam]
+          - variable: NAME
+            values: [Create]
       - variable: DATA_TYPE
         detection: python_shared_lang_datatype
         scope: result