diff --git a/TEST_MAPPING b/TEST_MAPPING
index 7200313..010eddf 100644
--- a/TEST_MAPPING
+++ b/TEST_MAPPING
@@ -1,103 +1,26 @@
 // Generated by update_crate_tests.py for tests that depend on this crate.
 {
-  "presubmit": [
+  "imports": [
     {
-      "name": "apkdmverity.test"
+      "path": "packages/modules/Virtualization/apkdmverity"
     },
     {
-      "name": "authfs_device_test_src_lib"
+      "path": "packages/modules/Virtualization/authfs"
     },
     {
-      "name": "avmdtool.test"
+      "path": "packages/modules/Virtualization/avmd"
     },
     {
-      "name": "avmdtool_tests"
+      "path": "packages/modules/Virtualization/libs/apkverify"
     },
     {
-      "name": "compos_verify.test"
+      "path": "packages/modules/Virtualization/microdroid_manager"
     },
     {
-      "name": "composd_cmd.test"
+      "path": "packages/modules/Virtualization/virtualizationmanager"
     },
     {
-      "name": "fd_server.test"
-    },
-    {
-      "name": "keystore2_client_tests"
-    },
-    {
-      "name": "libapkverify.integration_test"
-    },
-    {
-      "name": "libapkverify.test"
-    },
-    {
-      "name": "libpvmfw_avb.integration_test"
-    },
-    {
-      "name": "microdroid_manager_test"
-    },
-    {
-      "name": "rialto_test"
-    },
-    {
-      "name": "virtualizationmanager_device_test"
-    },
-    {
-      "name": "vm.test"
-    },
-    {
-      "name": "vmbase_example.integration_test"
-    }
-  ],
-  "presubmit-rust": [
-    {
-      "name": "apkdmverity.test"
-    },
-    {
-      "name": "authfs_device_test_src_lib"
-    },
-    {
-      "name": "avmdtool.test"
-    },
-    {
-      "name": "avmdtool_tests"
-    },
-    {
-      "name": "compos_verify.test"
-    },
-    {
-      "name": "composd_cmd.test"
-    },
-    {
-      "name": "fd_server.test"
-    },
-    {
-      "name": "keystore2_client_tests"
-    },
-    {
-      "name": "libapkverify.integration_test"
-    },
-    {
-      "name": "libapkverify.test"
-    },
-    {
-      "name": "libpvmfw_avb.integration_test"
-    },
-    {
-      "name": "microdroid_manager_test"
-    },
-    {
-      "name": "rialto_test"
-    },
-    {
-      "name": "virtualizationmanager_device_test"
-    },
-    {
-      "name": "vm.test"
-    },
-    {
-      "name": "vmbase_example.integration_test"
+      "path": "packages/modules/Virtualization/vm"
     }
   ]
 }