diff --git a/neuralnetworks/TEST_MAPPING b/neuralnetworks/TEST_MAPPING
index 421922a25e..0453a662ae 100644
--- a/neuralnetworks/TEST_MAPPING
+++ b/neuralnetworks/TEST_MAPPING
@@ -7,7 +7,7 @@
           // Just use sample-all driver for presubmit tests for faster results.
           // The other sample drivers (fast-float, quant, etc.) are subsets of
           // sample-all.
-          "native-test-flag": "--gtest_filter=*sample_all*"
+          "include-filter": "*sample_all*"
         }
       ]
     },
@@ -18,7 +18,7 @@
           // Just use sample-all driver for presubmit tests for faster results.
           // The other sample drivers (fast-float, quant, etc.) are subsets of
           // sample-all.
-          "native-test-flag": "--gtest_filter=*sample_all*"
+          "include-filter": "*sample_all*"
         }
       ]
     },
@@ -29,7 +29,7 @@
           // Just use sample-all driver for presubmit tests for faster results.
           // The other sample drivers (fast-float, quant, etc.) are subsets of
           // sample-all.
-          "native-test-flag": "--gtest_filter=*sample_all*"
+          "include-filter": "*sample_all*"
         }
       ]
     }