diff --git a/TEST_MAPPING b/TEST_MAPPING
index e1cd3ff..76a1dc0 100644
--- a/TEST_MAPPING
+++ b/TEST_MAPPING
@@ -1,16 +1,15 @@
 // Generated by update_crate_tests.py for tests that depend on this crate.
 {
+  "imports": [
+    {
+      "path": "system/security/keystore2"
+    },
+    {
+      "path": "system/security/keystore2/legacykeystore"
+    }
+  ],
   "presubmit": [
     {
-      "name": "keystore2_legacy_blobs_test"
-    },
-    {
-      "name": "keystore2_test"
-    },
-    {
-      "name": "legacykeystore_test"
-    },
-    {
       "name": "libsqlite3-sys_test_src_lib"
     },
     {
@@ -19,15 +18,6 @@
   ],
   "presubmit-rust": [
     {
-      "name": "keystore2_legacy_blobs_test"
-    },
-    {
-      "name": "keystore2_test"
-    },
-    {
-      "name": "legacykeystore_test"
-    },
-    {
       "name": "libsqlite3-sys_test_src_lib"
     },
     {