diff --git a/extension/ref-inject-code-search.js b/extension/ref-inject-code-search.js
index a09d96b..7e63c27 100644
--- a/extension/ref-inject-code-search.js
+++ b/extension/ref-inject-code-search.js
@@ -89,6 +89,9 @@ var _PACKAGE_MAP = {
   'android.sax'                          : { project:'platform/frameworks/base',     tree:'sax/java' },
   'android.telecom'                      : { project:'platform/frameworks/base',     tree:'telecomm/java' },
   'android.telephony'                    : { project:'platform/frameworks/base',     tree:'telephony/java' },
+  'android.test'                         : { project:'platform/frameworks/base',     tree:'test-runner/src' },
+  'android.test.mock'                    : { project:'platform/frameworks/base',     tree:'test-mock/src' },
+  'android.test.suitebuilder'            : { project:'platform/frameworks/base',     tree:'legacy-test/src' },
   'android.net.rtp'                      : { project:'platform/frameworks/opt/net/voip',      tree:'src/java' },
   'android.net.sip'                      : { project:'platform/frameworks/opt/net/voip',      tree:'src/java' },
   'android.net.wifi'                     : { project:'platform/frameworks/base',     tree:'wifi/java' },