diff --git a/update_source.sh b/update_source.sh
index ff7cc66..fc0c053 100755
--- a/update_source.sh
+++ b/update_source.sh
@@ -23,13 +23,18 @@
 SOURCE="https://github.com/linkedin/dexmaker"
 INCLUDE="
     LICENSE
+
     dexmaker
     dexmaker-mockito
-    dexmaker-mockito-tests
     dexmaker-mockito-inline
+    dexmaker-mockito-inline-extended
+
     dexmaker-mockito-inline-dispatcher
-    dexmaker-mockito-inline-tests/src
-    dexmaker-tests/src
+
+    dexmaker-tests
+    dexmaker-mockito-tests
+    dexmaker-mockito-inline-tests
+    dexmaker-mockito-inline-extended-tests
     "
 
 EXCLUDE="
@@ -58,11 +63,6 @@
   rm -r $exclude
 done;
 
-# Move the dexmaker-tests AndroidManifest.xml into the correct position.
-mv dexmaker-tests/src/main/AndroidManifest.xml dexmaker-tests/AndroidManifest.xml
-mv dexmaker-mockito-tests/src/main/AndroidManifest.xml dexmaker-mockito-tests/AndroidManifest.xml
-mv dexmaker-mockito-inline-tests/src/main/AndroidManifest.xml dexmaker-mockito-inline-tests/AndroidManifest.xml
-
 # Remove 3rd party code
 rm -r dexmaker-mockito-inline/external